| import gradio as gr | |
| from transformers import pipeline | |
| # Load the summarization pipeline | |
| pipe = pipeline("summarization", model="yashugupta786/bart_large_xsum_samsum_conv_summarizer") | |
| # Summarization function | |
| def summarize(text): | |
| summary = pipe(text) | |
| return summary[0]['summary_text'] | |
| # Gradio interface | |
| with gr.Blocks() as demo: | |
| gr.Interface( | |
| fn=summarize, | |
| inputs=gr.Textbox(label="Input Text", placeholder="Enter text to summarize..."), | |
| outputs=gr.Textbox(label="Summary"), | |
| title="Text Summarization", | |
| description="Summarize the input text using BART model." | |
| ).launch() | |