Update app.py
Browse files
app.py
CHANGED
|
@@ -165,8 +165,8 @@ def translate_batch(text_input):
|
|
| 165 |
with torch.no_grad():
|
| 166 |
outputs = model.generate(
|
| 167 |
**inputs,
|
| 168 |
-
max_length=
|
| 169 |
-
num_beams=
|
| 170 |
early_stopping=True
|
| 171 |
)
|
| 172 |
# output_ids shape: [batch_size, sequence_length]
|
|
|
|
| 165 |
with torch.no_grad():
|
| 166 |
outputs = model.generate(
|
| 167 |
**inputs,
|
| 168 |
+
max_length=1024,
|
| 169 |
+
num_beams=8,
|
| 170 |
early_stopping=True
|
| 171 |
)
|
| 172 |
# output_ids shape: [batch_size, sequence_length]
|