bart_summarize / app.py
TiffanySMY's picture
Upload app.py
4dc5e30 verified
raw
history blame contribute delete
812 Bytes
import streamlit as st
# Pipeline
from transformers import pipeline
summarizer = pipeline(task="summarization", model="TiffanySMY/Bart_model_for_summarization")
# Title
st.title("📝 Paragraph Summarization Using BART")
# Creating a sidebar for input
st.header("Input")
input_text = st.text_area("Enter a paragraph 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 paragraph for summarization.")