Jingshu commited on
Commit
7c406db
·
verified ·
1 Parent(s): 0cec5bc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -222,6 +222,7 @@ def translate_with_model(model_name, text, tgt_lang, src_lang, domain):
222
  max_new_tokens=500,
223
  past_key_values=past_key_values,
224
  streamer=streamer,
 
225
  )
226
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
227
  thread.start()
 
222
  max_new_tokens=500,
223
  past_key_values=past_key_values,
224
  streamer=streamer,
225
+ eos_token_id=TOKENIZER.eos_token_id,
226
  )
227
  thread = Thread(target=model.generate, kwargs=generation_kwargs)
228
  thread.start()