File size: 875 Bytes
d738ddb
d419e67
 
 
d738ddb
 
 
 
 
d419e67
 
069e418
d419e67
 
d738ddb
 
 
 
d419e67
d738ddb
 
 
 
 
 
 
d419e67
 
 
 
 
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
27
28
29
30
31
32
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.")