Update app.py

app.py

@@ -1,9 +1,23 @@
-import streamlit as st 
+import streamlit as st
 from transformers import pipeline
 
-# get the topic
-topic = st.text_area("Enter text to be summarized")
+# Load the summarization pipeline from Hugging Face
+summarizer = pipeline("summarization", model="sshleifer/distilbart-cnn-12-6")
 
-if topic:
-    summarizer = pipeline("summarization", model = "google-t5/t5-base")
-    st.json(summarizer)
+# Streamlit interface
+st.title("Text Summarization App")
+
+# Text input
+input_text = st.text_area("Enter the text to summarize:", height=200)
+
+if st.button("Summarize"):
+    if input_text:
+        # Summarize the text
+        summary = summarizer(input_text, max_length=50, min_length=25, do_sample=False)
+        summarized_text = summary[0]['summary_text']
+        
+        # Display the summary
+        st.subheader("Summary")
+        st.write(summarized_text)
+    else:
+        st.error("Please enter some text to summarize.")