File size: 907 Bytes
67d23d6 397b5d5 67d23d6 f79f33f 67d23d6 126e812 67d23d6 38780dc 67d23d6 126e812 67d23d6 5e2190f 67d23d6 38780dc |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 |
import streamlit as st
from transformers import pipeline
# Instantiating summarization pipeline with the bart-finetuned-samsum model
summarizer = pipeline(task="summarization", model="Priyanka-Balivada/pegasus-samsum")
# Title
st.title("📝 Text Summarization with Pegasus")
# 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.")
|