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