Update app.py
Browse files
app.py
CHANGED
|
@@ -74,6 +74,8 @@ def model_inference(
|
|
| 74 |
|
| 75 |
thread = Thread(target=model.generate, kwargs=generation_args)
|
| 76 |
thread.start()
|
|
|
|
|
|
|
| 77 |
buffer = ""
|
| 78 |
for new_text in streamer:
|
| 79 |
try:
|
|
|
|
| 74 |
|
| 75 |
thread = Thread(target=model.generate, kwargs=generation_args)
|
| 76 |
thread.start()
|
| 77 |
+
thread.join()
|
| 78 |
+
|
| 79 |
buffer = ""
|
| 80 |
for new_text in streamer:
|
| 81 |
try:
|