Update app.py

app.py

@@ -54,8 +54,8 @@ def display_results(data):
 # Function to summarize text using the specified model
 def summarize_text(text):
     try:
-        # Initialize the summarization model
-        summarizer = pipeline("text2text-generation", model="spacemanidol/flan-t5-large-website-summarizer")
+        # Initialize the summarization model with PyTorch
+        summarizer = pipeline("text2text-generation", model="spacemanidol/flan-t5-large-website-summarizer", framework="pt")
         summary = summarizer(text, max_length=150, min_length=50, do_sample=False)
         return summary[0]['generated_text']
     except Exception as e:
@@ -65,8 +65,8 @@ def summarize_text(text):
 # Function to generate text (if you want to keep this)
 def generate_text(text):
     try:
-        # Initialize the text generation model
-        text_generator = pipeline("text2text-generation", model="JorgeSarry/est5-summarize")
+        # Initialize the text generation model with PyTorch
+        text_generator = pipeline("text2text-generation", model="JorgeSarry/est5-summarize", framework="pt")
         generated_text = text_generator(text, max_length=150, min_length=50, do_sample=False)
         return generated_text[0]['generated_text']
     except Exception as e: