Update app.py

app.py

@@ -1,63 +1,29 @@
 import streamlit as st
-from streamlit.components.v1 import html
 from transformers import BartForConditionalGeneration, BartTokenizer
 
-# Function to add custom CSS for background styling
-def set_background(style):
-    component = """
-        <style>
-            .stApp {
-                %s
-            }
-        </style>
-    """ % style
-    return html(component, height=0)
+def summarize_text(data, max_length):
+    model_name = "facebook/bart-large-cnn"
+    model = BartForConditionalGeneration.from_pretrained(model_name)
+    tokenizer = BartTokenizer.from_pretrained(model_name)
 
-def summarize_data(data, max_length, model, tokenizer):
-    inputs = tokenizer.encode("summarize: " + data, return_tensors="pt", max_length=1024, truncation=True)
-    summary_ids = model.generate(inputs, max_length=max_length, min_length=max_length//4, length_penalty=2.0, num_beams=4, early_stopping=True)
+    inputs = tokenizer(data, max_length=max_length, return_tensors="pt", truncation=True)
+    summary_ids = model.generate(inputs["input_ids"])
 
-    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
-    return summary
+    summary_text = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary_text
 
 def main():
-    # Set background style
-    set_background("""
-        background-image: url('https://your-image-url.jpg');
-        background-size: cover;
-    """)
-
-    # Title and description
     st.title("Data Summarization")
-    st.write("Enter your data, set the summary length, and click submit.")
-
-    # Text input box
-    data = st.text_area("Enter your Data", height=200)
 
-    # Summary Length slider
-    max_length = st.slider("Summary Length", min_value=20, max_value=1000, step=1)
-
-    # Load the pretrained BART model and tokenizer
-    model_name = "facebook/bart-large-cnn"
-    model = BartForConditionalGeneration.from_pretrained(model_name)
-    tokenizer = BartTokenizer.from_pretrained(model_name)
+    data = st.text_area("Enter your Data", "")
+    max_length = st.slider("Summary Length", 20, 1000, 200)
 
-    # Submit button
     if st.button("Submit"):
         if not data:
-            st.warning("Please enter some data.")
+            st.warning("Please enter some text for summarization.")
         else:
-            # Summarize data
-            result = summarize_data(data, max_length, model, tokenizer)
-            
-            # Display result in a stylized output box
-            st.markdown(
-                f"<div style='background-color:#8c88f9; padding: 10px; border-radius: 10px;'>"
-                f"<h3>Your Summary</h3>"
-                f"<p>{result}</p>"
-                f"</div>",
-                unsafe_allow_html=True,
-            )
+            summary = summarize_text(data, max_length)
+            st.text_area("Your Summary", summary, height=200)
 
 if __name__ == "__main__":
-    main()
+    main()