Create app.py

app.py

@@ -0,0 +1,63 @@
+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_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)
+
+    summary = tokenizer.decode(summary_ids[0], skip_special_tokens=True)
+    return summary
+
+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)
+
+    # Submit button
+    if st.button("Submit"):
+        if not data:
+            st.warning("Please enter some data.")
+        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,
+            )
+
+if __name__ == "__main__":
+    main()