Spaces:
Sleeping
Sleeping
import streamlit as st | |
from transformers import pipeline, logging | |
# Suppress warnings from transformers | |
logging.set_verbosity_error() | |
# Streamlit interface setup | |
st.title("Summarization Test") | |
# Initialize the summarizer | |
summarizer = pipeline("summarization") | |
# Test summarization with a hardcoded string | |
test_text = "This is a simple test sentence to verify the functionality of the summarization pipeline. The goal is to ensure that the pipeline can process input text correctly and produce a summary without encountering the input type error." | |
if st.button('Test Summarization'): | |
try: | |
# Attempt to summarize the hardcoded test text | |
summary = summarizer(test_text, max_length=130, min_length=30, do_sample=False) | |
st.success("Summarization succeeded!") | |
st.write("### Summary:") | |
st.write(summary[0]['summary_text']) | |
except Exception as e: | |
st.error(f"Summarization test failed: {e}") | |