Create app.py

app.py

@@ -0,0 +1,25 @@
+import streamlit as st
+from transformers import pipeline
+
+# Load the model from Hugging Face
+@st.cache_resource
+def load_model():
+    summarization_pipeline = pipeline("summarization", model="tareeb23/summarizer_bart")
+    return summarization_pipeline
+
+summarization_pipeline = load_model()
+
+# Streamlit app
+st.title("Dialogue Summarizer")
+st.write("Enter a dialogue to get a summarized version.")
+
+# User input
+dialogue = st.text_area("Enter Dialogue", "")
+
+if st.button("Summarize"):
+    if dialogue:
+        summary = summarization_pipeline(dialogue)
+        st.write("### Summary:")
+        st.write(summary[0]['summary_text'])
+    else:
+        st.write("Please enter a dialogue to summarize.")