Update README.md
Browse files
README.md
CHANGED
|
@@ -76,7 +76,7 @@ def summarize_article(article, model, tokenizer):
|
|
| 76 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
| 77 |
return summary
|
| 78 |
|
| 79 |
-
# Load
|
| 80 |
t5_model_custom = T5ForConditionalGeneration.from_pretrained("Vijayendra/T5-Base-Sum")
|
| 81 |
t5_tokenizer_custom = T5Tokenizer.from_pretrained("Vijayendra/T5-Base-Sum")
|
| 82 |
|
|
@@ -105,7 +105,7 @@ misinformation policies on YouTube with new guidelines on currently administered
|
|
| 105 |
effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
|
| 106 |
"""
|
| 107 |
|
| 108 |
-
# Summarize with
|
| 109 |
t5_summary_custom = summarize_article(article, t5_model_custom, t5_tokenizer_custom)
|
| 110 |
|
| 111 |
# Summarize with the pretrained T5 model for summarization
|
|
|
|
| 76 |
summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
|
| 77 |
return summary
|
| 78 |
|
| 79 |
+
# Load our fine-tuned T5 model and tokenizer
|
| 80 |
t5_model_custom = T5ForConditionalGeneration.from_pretrained("Vijayendra/T5-Base-Sum")
|
| 81 |
t5_tokenizer_custom = T5Tokenizer.from_pretrained("Vijayendra/T5-Base-Sum")
|
| 82 |
|
|
|
|
| 105 |
effective by local health authorities and the WHO," the post said, referring to the World Health Organization.
|
| 106 |
"""
|
| 107 |
|
| 108 |
+
# Summarize with our fine-tuned T5 model
|
| 109 |
t5_summary_custom = summarize_article(article, t5_model_custom, t5_tokenizer_custom)
|
| 110 |
|
| 111 |
# Summarize with the pretrained T5 model for summarization
|