mgoin commited on
Commit
ef8a4e8
·
1 Parent(s): 2e6fcd5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -112,7 +112,7 @@ with gr.Blocks() as demo:
112
  "max_new_tokens": max_new_tokens,
113
  "temperature": temperature,
114
  }
115
- inference = pipe(sequences=message, streaming=False, **generation_config)
116
  history[-1][1] += message
117
  for token in inference:
118
  history[-1][1] += token.generations[0].text
 
112
  "max_new_tokens": max_new_tokens,
113
  "temperature": temperature,
114
  }
115
+ inference = pipe(sequences=message, streaming=True, **generation_config)
116
  history[-1][1] += message
117
  for token in inference:
118
  history[-1][1] += token.generations[0].text