Yaron Koresh commited on
Commit
aa7cc5b
·
verified ·
1 Parent(s): 966bdcc

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -565,6 +565,8 @@ def summarize_text(
565
  print(f'DBG summarize_text 1 {i}')
566
  outputs = model.generate(
567
  inputs[:512],
 
 
568
  length_penalty=2.0,
569
  num_beams=4,
570
  early_stopping=True
 
565
  print(f'DBG summarize_text 1 {i}')
566
  outputs = model.generate(
567
  inputs[:512],
568
+ max_length=max_len,
569
+ min_length=min_len,
570
  length_penalty=2.0,
571
  num_beams=4,
572
  early_stopping=True