sara musaeva
commited on
Commit
·
036696c
1
Parent(s):
2bbbf23
Update app.py
Browse files
app.py
CHANGED
|
@@ -13,7 +13,7 @@ def translate_to_russian(input_text):
|
|
| 13 |
outputs = model.generate(
|
| 14 |
input_ids=input_ids,
|
| 15 |
max_length=256,
|
| 16 |
-
|
| 17 |
no_repeat_ngram_size=2,
|
| 18 |
length_penalty=1.2
|
| 19 |
)
|
|
|
|
| 13 |
outputs = model.generate(
|
| 14 |
input_ids=input_ids,
|
| 15 |
max_length=256,
|
| 16 |
+
num_beams=3,
|
| 17 |
no_repeat_ngram_size=2,
|
| 18 |
length_penalty=1.2
|
| 19 |
)
|