Spaces:
Paused
Paused
Update app.py
Browse files
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()
|