Spaces:
Paused
Paused
Update app.py
Browse files
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=
|
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
|