|
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() |