Last commit not found
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() | |