Update app.py

app.py

@@ -1,5 +1,7 @@
+import streamlit as st
 from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
 
+# Load the tokenizer and model
 tokenizer = AutoTokenizer.from_pretrained("suriya7/bart-finetuned-text-summarization")
 model = AutoModelForSeq2SeqLM.from_pretrained("suriya7/bart-finetuned-text-summarization")
 
@@ -9,9 +11,20 @@ def generate_summary(text):
     summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
     return summary
 
-text_to_summarize = """Now, there is no doubt that one of the most important aspects of any Pixel phone is its camera.
-And there might be good news for all camera lovers. Rumours have suggested that the Pixel 9 could come with a telephoto lens,
-improving its photography capabilities even further. Google will likely continue to focus on using AI to enhance its camera performance,
-in order to make sure that Pixel phones remain top contenders in the world of mobile photography."""
-summary = generate_summary(text_to_summarize)
-print(summary)
+# Streamlit interface
+st.title("Text Summarization App")
+
+# User text input
+user_input = st.text_area("Enter the text you want to summarize", height=200)
+
+if st.button("Generate Summary"):
+    if user_input:
+        with st.spinner("Generating summary..."):
+            summary = generate_summary(user_input)
+        st.subheader("Summary:")
+        st.write(summary)
+    else:
+        st.warning("Please enter text to summarize.")
+
+# Instructions for using the app
+st.write("Enter your text in the box above and click 'Generate Summary' to get a summarized version of your text.")