MahmoudH commited on
Commit
ec5ce8f
·
1 Parent(s): dd9ff49

Increase `num_beams`

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -12,7 +12,7 @@ def predict(text):
12
  attention_mask=tokenized_inputs["attention_mask"],
13
  max_new_tokens=256,
14
  length_penalty=0.8,
15
- num_beams=2,
16
  do_sample=True
17
  )
18
  summary = tokenizer.batch_decode(output, skip_special_tokens=True)[0]
 
12
  attention_mask=tokenized_inputs["attention_mask"],
13
  max_new_tokens=256,
14
  length_penalty=0.8,
15
+ num_beams=4,
16
  do_sample=True
17
  )
18
  summary = tokenizer.batch_decode(output, skip_special_tokens=True)[0]