hardik90's picture
Update app.py
d82099c verified
raw
history blame contribute delete
778 Bytes
import streamlit as st
from transformers import pipeline
# Load summarization model
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
# Streamlit App
st.title("Summarization App")
# Input text area for user to enter the article
article = st.text_area("Enter the article:", "", height=300)
if st.button("Generate Summary"):
if article:
# Generate summary using the summarizer pipeline
summary_result = summarizer(article, max_length=130, min_length=30, do_sample=False)
# Extract the summary text
summary_text = summary_result[0]['summary_text']
# Display the summary
st.subheader("Summary:")
st.write(summary_text)
else:
st.warning("Please enter an article for summarization.")