increase max length abs summarizer
Browse files
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=
|
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)
|