ysharma HF staff commited on
Commit
4a0f685
1 Parent(s): c0bcc27
Files changed (1) hide show
  1. app.py +3 -2
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 = ""