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.") | |