Yaron Koresh commited on
Commit
d87eb53
·
verified ·
1 Parent(s): 9835438

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -592,8 +592,8 @@ def _summarize(text):
592
  toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
593
  gen = model.generate(
594
  toks,
595
- length_penalty=0.9,
596
- num_beams=8,
597
  early_stopping=True,
598
  max_length=512
599
  )
 
592
  toks = tokenizer.encode( prefix + text, return_tensors="pt", truncation=False)
593
  gen = model.generate(
594
  toks,
595
+ length_penalty=2.0,
596
+ num_beams=4,
597
  early_stopping=True,
598
  max_length=512
599
  )