Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
@@ -62,6 +62,7 @@ def generate(prompt, temperature, top_p, top_k, repetition_penalty, max_tokens,
|
|
62 |
if event.is_set():
|
63 |
raise TimeoutError("[ERROR] Generation timed out.")
|
64 |
buffer += item["choices"][0]["text"]
|
|
|
65 |
yield buffer
|
66 |
finally:
|
67 |
timer.cancel()
|
|
|
62 |
if event.is_set():
|
63 |
raise TimeoutError("[ERROR] Generation timed out.")
|
64 |
buffer += item["choices"][0]["text"]
|
65 |
+
print(item)
|
66 |
yield buffer
|
67 |
finally:
|
68 |
timer.cancel()
|