Update app.py

app.py

@@ -1,16 +1,16 @@
 import streamlit as st
 from transformers import pipeline
 
-# Load the summarization pipeline
-summarizer = pipeline("summarization")
+# Load the T5 summarization pipeline
+summarizer = pipeline("summarization", model="t5-small", tokenizer="t5-small")
 
 def summarize_text(text):
-    """Summarize the input text using Hugging Face's pipeline."""
+    """Summarize the input text using the T5 model."""
     summary = summarizer(text, max_length=150, min_length=50, do_sample=False)
     return summary[0]['summary_text']
 
 # Streamlit UI
-st.title("Text Summarization with Hugging Face")
+st.title("Text Summarization with T5")
 
 st.write("Enter the text you want to summarize:")
 
@@ -25,4 +25,3 @@ if st.button("Summarize"):
         st.write(summary)
     else:
         st.error("Please enter some text to summarize.")
-