File size: 625 Bytes
f119c21 3b60922 f119c21 9deb7ac | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 | import gradio as gr
from transformers import pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
# Function to summarize text
def summarize_text(text):
summary = summarizer(text, max_length=100, min_length=30, do_sample=False)
return summary[0]['summary_text']
# Create Gradio UI
iface = gr.Interface(
fn=summarize_text,
inputs=gr.Textbox(lines=5, placeholder="Enter text to summarize..."),
outputs="text",
title="Text Summarization Chatbot",
description="Enter a long text, and the chatbot will generate a concise summary using BART."
)
iface.launch(share=True) |