Yaron Koresh commited on
Commit
e8cd0ee
·
verified ·
1 Parent(s): 27d4615

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -566,7 +566,7 @@ def _summarize(text, max_words=20):
566
  gen = model.generate(
567
  toks,
568
  length_penalty=2.0,
569
- num_beams=max( get_tensor_length(toks) // 4 , 4 ),
570
  early_stopping=True,
571
  max_length=max_words
572
  )
 
566
  gen = model.generate(
567
  toks,
568
  length_penalty=2.0,
569
+ num_beams=4,
570
  early_stopping=True,
571
  max_length=max_words
572
  )