destiratnakomala
commited on
Commit
•
b70a0c8
1
Parent(s):
6db5daf
Update README.md
Browse files
README.md
CHANGED
@@ -39,9 +39,9 @@ ARTICLE_TO_SUMMARIZE = ""
|
|
39 |
# generate summary
|
40 |
input_ids = tokenizer.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
|
41 |
summary_ids = model.generate(input_ids,
|
42 |
-
min_length=
|
43 |
-
max_length=
|
44 |
-
num_beams=
|
45 |
repetition_penalty=2.5,
|
46 |
length_penalty=1.0,
|
47 |
early_stopping=True,
|
|
|
39 |
# generate summary
|
40 |
input_ids = tokenizer.encode(ARTICLE_TO_SUMMARIZE, return_tensors='pt')
|
41 |
summary_ids = model.generate(input_ids,
|
42 |
+
min_length=200,
|
43 |
+
max_length=800,
|
44 |
+
num_beams=100,
|
45 |
repetition_penalty=2.5,
|
46 |
length_penalty=1.0,
|
47 |
early_stopping=True,
|