Spaces:
Build error
Build error
upd
Browse files
app.py
CHANGED
|
@@ -38,8 +38,9 @@ def predict(message, history):
|
|
| 38 |
num_beams=1,
|
| 39 |
stopping_criteria=StoppingCriteriaList([stop])
|
| 40 |
)
|
| 41 |
-
t = Thread(target=model.generate, kwargs=generate_kwargs)
|
| 42 |
-
t.start()
|
|
|
|
| 43 |
|
| 44 |
#Initialize an empty string to store the generated text
|
| 45 |
partial_message = ""
|
|
|
|
| 38 |
num_beams=1,
|
| 39 |
stopping_criteria=StoppingCriteriaList([stop])
|
| 40 |
)
|
| 41 |
+
#t = Thread(target=model.generate, kwargs=generate_kwargs)
|
| 42 |
+
#t.start()
|
| 43 |
+
model.generate(**generate_kwargs)
|
| 44 |
|
| 45 |
#Initialize an empty string to store the generated text
|
| 46 |
partial_message = ""
|