Zain4s's picture
Update app.py
069e418 verified
raw
history blame
875 Bytes
import streamlit as st
from transformers import pipeline
summarizer = pipeline("summarization", model="bart_base_samsum")
# Function to clear text
def clear_text():
st.session_state["input_text"] = ""
# Function to summarize text
def summarize_text(text):
summary = summarizer(text, max_length=100, min_length=10, do_sample=False)[0]['summary_text']
return summary
st.title("Text Summarizer")
input_text = st.text_area("Input", value="", key="input_text", height=300)
# Clear & Submit bittons
col1, col2 = st.columns([1, 1])
with col1:
if st.button("Clear", on_click=clear_text):
pass
with col2:
if st.button("Submit"):
if input_text.strip():
summary = summarize_text(input_text)
st.text_area("Summary", value=summary, height=300)
else:
st.warning("Please enter some text to summarize.")