Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -61,7 +61,7 @@ def generate_output(task, source, target, input_text, mt_text=None):
|
|
| 61 |
input_ids=inputs.input_ids,
|
| 62 |
max_new_tokens=4000,
|
| 63 |
early_stopping=True,
|
| 64 |
-
num_beams=
|
| 65 |
)
|
| 66 |
|
| 67 |
decoded = tokenizer.decode(output[0, input_length:], skip_special_tokens=True).strip()
|
|
|
|
| 61 |
input_ids=inputs.input_ids,
|
| 62 |
max_new_tokens=4000,
|
| 63 |
early_stopping=True,
|
| 64 |
+
num_beams=1
|
| 65 |
)
|
| 66 |
|
| 67 |
decoded = tokenizer.decode(output[0, input_length:], skip_special_tokens=True).strip()
|