Update app.py

app.py

@@ -1,29 +1,33 @@
 import streamlit as st
-from transformers import BartTokenizer, BartForSequenceClassification
+from transformers import pipeline
 
-def summarize_text(data, max_length):
-    model_name = "facebook/bart-large-cnn"
-    model = BartForSequenceClassification.from_pretrained(model_name)
-    tokenizer = BartTokenizer.from_pretrained(model_name)
-
-    inputs = tokenizer(data, max_length=max_length, return_tensors="pt", truncation=True)
-    summary_ids = model.generate(inputs["input_ids"])
-
-    summary_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-    return summary_text
+# Load the summarization model from Hugging Face
+summarizer = pipeline("summarization")
 
+# Streamlit app
 def main():
-    st.title("Data Summarization")
+    # Set the title and description of the app
+    st.title("Text Summarizer App")
+    st.write(
+        "Enter a piece of text, select the length of the summary, and get a concise summary!"
+    )
+
+    # Input text box for user input
+    input_text = st.text_area("Enter your text here:")
 
-    data = st.text_area("Enter your Data", "")
-    max_length = st.slider("Summary Length", 20, 1000, 200)
+    # Length selector for summary
+    summary_length = st.slider("Select summary length:", min_value=50, max_value=500, value=150, step=50)
 
-    if st.button("Submit"):
-        if not data:
-            st.warning("Please enter some text for summarization.")
+    # Check if the user has entered any text
+    if st.button("Generate Summary"):
+        if not input_text:
+            st.warning("Please enter some text.")
         else:
-            summary = summarize_text(data, max_length)
-            st.text_area("Your Summary", summary, height=200)
+            # Generate summary using the Hugging Face summarization model
+            summary = summarizer(input_text, max_length=summary_length, min_length=50, length_penalty=2.0, num_beams=4)
+            st.subheader("Summary:")
+            st.write(summary[0]["summary_text"])
 
+# Run the app
 if __name__ == "__main__":
     main()