Spaces:
Sleeping
Sleeping
| import gradio as gr | |
| from transformers import pipeline | |
| # Load the summarization pipeline | |
| summarizer = pipeline("summarization", model="facebook/bart-large-cnn") | |
| # Define the summarization function | |
| def summarize_text(input_text): | |
| if not input_text.strip(): | |
| return "Please provide some text to summarize." | |
| try: | |
| summary = summarizer(input_text, max_length=130, min_length=30, do_sample=False) | |
| return summary[0]["summary_text"] | |
| except Exception as e: | |
| return f"Error: {str(e)}" | |
| # Create the Gradio interface | |
| demo = gr.Interface( | |
| fn=summarize_text, | |
| inputs=gr.Textbox(lines=5, label="Input Text"), | |
| outputs=gr.Textbox(label="Summarized Text"), | |
| title="Text Summarizer", | |
| description="Summarize long articles or paragraphs using the `facebook/bart-large-cnn` model." | |
| ) | |
| # Launch the app | |
| if __name__ == "__main__": | |
| demo.launch() | |