BenBranyon commited on
Commit
ae1188f
·
verified ·
1 Parent(s): 3814e21

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -92,7 +92,7 @@ def generate(
92
  num_beams=1,
93
  repetition_penalty=repetition_penalty,
94
  )
95
- t = Thread(target=model.generate)
96
  t.start()
97
 
98
  outputs = []
 
92
  num_beams=1,
93
  repetition_penalty=repetition_penalty,
94
  )
95
+ t = Thread(target=model.generate, kwargs=generate_kwargs)
96
  t.start()
97
 
98
  outputs = []