Vijayendra
commited on
Update README.md
Browse files
README.md
CHANGED
@@ -37,7 +37,7 @@ misinformation policies on YouTube with new guidelines on currently administered
|
|
37 |
effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
|
38 |
"""
|
39 |
inputs = tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=512, truncation=True)
|
40 |
-
summary_ids = model.generate(inputs, max_length=150, min_length=
|
41 |
|
42 |
# Decode and print the summary
|
43 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
|
|
37 |
effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
|
38 |
"""
|
39 |
inputs = tokenizer.encode("summarize: " + article, return_tensors="pt", max_length=512, truncation=True)
|
40 |
+
summary_ids = model.generate(inputs, max_length=150, min_length=100, length_penalty=2.0, num_beams=4, early_stopping=True)
|
41 |
|
42 |
# Decode and print the summary
|
43 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|