Update app.py

app.py

@@ -19,7 +19,8 @@ from transformers import pipeline
 
 # from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
-from transformers import AutoTokenizer, PegasusXForConditionalGeneration
+from transformers import T5Tokenizer, T5ForConditionalGeneration
+
 
 def search_related_articles_crossref(query, max_results=3):
     """Search for related articles using CrossRef API."""
@@ -64,8 +65,9 @@ def extract_text_from_html(url):
     except Exception as e:
         return f"Error extracting text: {str(e)}"
 
-model = PegasusXForConditionalGeneration.from_pretrained("google/pegasus-x-base")
-tokenizer = AutoTokenizer.from_pretrained("google/pegasus-x-large")
+
+tokenizer = T5Tokenizer.from_pretrained("google-t5/t5-small")
+model = T5ForConditionalGeneration.from_pretrained("google-t5/t5-small")
 
 def summarize_article(article_text):
     """Summarize a given article's text."""