Vijayendra
commited on
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
|