Spaces:
Sleeping
Sleeping
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.") | |