A2 / app.py
shahad-b's picture
Update app.py
aacb8ea verified
raw
history blame contribute delete
694 Bytes
import gradio as gr
from transformers import pipeline
sum = pipeline("summarization", model="facebook/bart-large-cnn")
def sumtext(text, min, max):
summary = sum(text, min_length=min, max_length=max, do_sample=False)
return summary[0]['summary']
gr = gr.Interface(
fn=sumtext,
inputs=[
gr.Textbox(label="Enter Text", lines=10),
gr.Slider(minimum=10, maximum=50, label="Minimum Summary Length", value=10),
gr.Slider(minimum=50, maximum=150, label="Maximum Summary Length", value=100)
],
outputs=gr.Textbox(label="Summarized Text"),
title="Text Summarizer",
description="Provide a lengthy text to receive a brief summary."
)
gr.launch()