Gladiator commited on
Commit
4c24ae0
1 Parent(s): 8c80b2b

increase max length abs summarizer

Browse files
Files changed (1) hide show
  1. src/abstractive_summarizer.py +1 -1
src/abstractive_summarizer.py CHANGED
@@ -14,7 +14,7 @@ def abstractive_summarizer(tokenizer, model, text):
14
  num_beams=4,
15
  no_repeat_ngram_size=2,
16
  min_length=30,
17
- max_length=100,
18
  early_stopping=True,
19
  )
20
  abs_summarized_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
 
14
  num_beams=4,
15
  no_repeat_ngram_size=2,
16
  min_length=30,
17
+ max_length=300,
18
  early_stopping=True,
19
  )
20
  abs_summarized_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)