Upload 3 files

README.md

@@ -1,12 +1,12 @@
 ---
-title: Text
-emoji: 👀
-colorFrom: yellow
-colorTo: yellow
+title: Bart Text Summarization
+emoji: 📝
+colorFrom: blue
+colorTo: purple
 sdk: streamlit
-sdk_version: 1.31.1
+sdk_version: 1.28.0
 app_file: app.py
 pinned: false
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference# text-summarization

app.py

@@ -0,0 +1,25 @@
+import streamlit as st
+from transformers import pipeline
+
+# Instantiating summarization pipeline with the bart-finetuned-samsum model
+summarizer = pipeline(task="summarization", model="luisotorres/bart-finetuned-samsum")
+
+# Title
+st.title("📝 Text Summarization with BART")
+
+# Creating a sidebar for input
+with st.sidebar:
+    st.header("Input")
+    input_text = st.text_area("Enter a text or dialogue for summarization.")
+
+# Creating a button to start the summarization
+if st.button("Summarize"):
+    # If the input box isn't empty, process the input and generate a summary
+    if input_text:
+        summary = summarizer(input_text, max_length=1024, min_length=0, do_sample=False)
+        st.subheader("Original Text")
+        st.write(input_text)
+        st.subheader("Summary")
+        st.write(summary[0]["summary_text"])
+    else:
+        st.warning("Enter a text or dialogue for summarization.")

requirements.txt

@@ -0,0 +1,3 @@
+transformers
+torch
+streamlit