Update app.py

app.py

@@ -4,14 +4,14 @@ from transformers import pipeline
 def main():
     st.title("Text Summarization")
 
-    # Initialize the summarizer pipeline
+    # Initialize the summarizer pipeline with a more powerful model
     summarizer = pipeline(
         task="summarization",
-        model="facebook/bart-large-cnn",  # Using a different model for better summarization
-        min_length=50,  # Increased minimum length to capture more details
-        max_length=150,  # Adjusted max length to allow for more detailed summaries
+        model="facebook/bart-large-cnn",  # Consider using a larger model
+        min_length=50,
+        max_length=150,
         truncation=True,
-    )
+    ) 
 
     # User input
     input_text = st.text_area("Enter the text you want to summarize:", height=200)
@@ -19,15 +19,22 @@ def main():
     # Summarize button
     if st.button("Summarize"):
         if input_text:
-            # Generate the summary
-            output = summarizer(input_text, max_length=150, min_length=50, do_sample=False)
-            summary = output[0]['summary_text']
+            # Split the text into smaller chunks if it's too long
+            max_input_length = 1024  # BART can handle up to 1024 tokens
+            input_chunks = [input_text[i:i+max_input_length] for i in range(0, len(input_text), max_input_length)]
+
+            # Generate the summary for each chunk and combine them
+            summary = ""
+            for chunk in input_chunks:
+                output = summarizer(chunk, max_length=150, min_length=50, do_sample=False)
+                summary += output[0]['summary_text'] + " "
 
             # Display the summary as bullet points
             st.subheader("Summary:")
             bullet_points = summary.split(". ")
             for point in bullet_points:
-                st.write(f"- {point.strip()}")
+                if point:  # Ensure that empty strings are not included
+                    st.write(f"- {point.strip()}")
         else:
             st.warning("Please enter text to summarize.")