Update app.py

app.py

@@ -14,10 +14,12 @@ import requests
 from transformers import pipeline
 
 # Initialize Hugging Face Summarization and Text Generation Pipelines
-summarizer = pipeline("summarization", model="scieditor/citation-generation-t5")
-citation_generator = pipeline("text-generation", model="gpt2")
+# summarizer = pipeline("summarization", model="scieditor/citation-generation-t5")
+# citation_generator = pipeline("text-generation", model="gpt2")
 
-from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+# from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
+
+from transformers import AutoTokenizer, PegasusModel
 
 def search_related_articles_crossref(query, max_results=3):
     """Search for related articles using CrossRef API."""
@@ -62,8 +64,8 @@ def extract_text_from_html(url):
     except Exception as e:
         return f"Error extracting text: {str(e)}"
 
-tokenizer = AutoTokenizer.from_pretrained("gpt2")
-model = AutoModelForSeq2SeqLM.from_pretrained("gpt2")
+tokenizer = AutoTokenizer.from_pretrained("google/pegasus-x-large")
+model = PegasusModel.from_pretrained("google/pegasus-x-large")
 
 def summarize_article(article_text):
     """Summarize a given article's text."""