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()
|