javi8979 commited on
Commit
bfdc52b
·
verified ·
1 Parent(s): 2873048

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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=5
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()