summerizer / app.py
nxmwxm's picture
Update app.py
f7f73d5 verified
raw
history blame
580 Bytes
!pip install torch
import torch
import streamlit as st
from transformers import pipeline
# Initialize the summarization pipeline
summarizer = pipeline("summarization")
st.title("Text Summarization Tool")
st.write("Enter the text you want to summarize below:")
# Text input
text = st.text_area("Text Input", "")
if st.button("Summarize"):
if text:
summary = summarizer(text, max_length=150, min_length=30, do_sample=False)
st.write("Summary:")
st.write(summary[0]['summary_text'])
else:
st.write("Please enter some text to summarize.")