Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline | |
# Function to extract text from SRT-formatted text | |
def extract_text_from_srt_text(srt_text): | |
lines = srt_text.strip().split("\n") | |
texts = [line.split("\n")[2] for line in lines if not line.startswith("0")] | |
return " ".join(texts) | |
# Load summarization pipeline | |
summarizer = pipeline("summarization") | |
# Streamlit app | |
st.title("SRT Summarization") | |
# Text area for user to input SRT-formatted text | |
srt_text_input = st.text_area("Paste SRT-formatted text here:") | |
# Button to trigger summarization | |
if st.button("Summarize"): | |
# Check if text area is not empty | |
if srt_text_input.strip(): | |
# Extract text from SRT-formatted text | |
text_to_summarize = extract_text_from_srt_text(srt_text_input) | |
# Summarize text | |
summary = summarizer(text_to_summarize, max_length=1500, min_length=10, do_sample=False) | |
# Display summary | |
st.subheader("Summary:") | |
st.write(summary[0]["summary_text"]) | |
else: | |
st.warning("Please enter some SRT-formatted text.") | |