navjotk's picture
Create app.py
1f1e254 verified
raw
history blame contribute delete
704 Bytes
import gradio as gr
from transformers import pipeline
# Load the summarization pipeline with facebook/bart-large-cnn
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
# Function to summarize text
def summarize_text(text):
summary = summarizer(text, max_length=130, min_length=30, do_sample=False)
return summary[0]['summary_text']
# Gradio Interface
demo = gr.Interface(
fn=summarize_text,
inputs=gr.Textbox(lines=10, label="Enter Text to Summarize"),
outputs=gr.Textbox(label="Summary"),
title="Text Summarizer using facebook/bart-large-cnn",
description="Paste any long text and get a short summary using BART model from Facebook."
)
demo.launch()