Spaces:
Paused
Paused
fix style
Browse files
app.py
CHANGED
|
@@ -51,11 +51,6 @@ def generate(
|
|
| 51 |
|
| 52 |
history = history_with_input[:-1]
|
| 53 |
generator = run(message, history, max_new_tokens, temperature, top_p, top_k)
|
| 54 |
-
try:
|
| 55 |
-
first_response = next(generator)
|
| 56 |
-
yield history + [(message, first_response)]
|
| 57 |
-
except StopIteration:
|
| 58 |
-
yield history + [(message, '')]
|
| 59 |
for response in generator:
|
| 60 |
yield history + [(message, response)]
|
| 61 |
|
|
|
|
| 51 |
|
| 52 |
history = history_with_input[:-1]
|
| 53 |
generator = run(message, history, max_new_tokens, temperature, top_p, top_k)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 54 |
for response in generator:
|
| 55 |
yield history + [(message, response)]
|
| 56 |
|