File size: 559 Bytes
6df46b0
 
 
 
01f683c
6df46b0
 
a7486bd
6df46b0
 
 
01f683c
 
 
 
6df46b0
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
import gradio as gr
from transformers import pipeline


def main(article, min_len, max_len):
    summarizer = pipeline("summarization", model="facebook/bart-large-cnn")

    return summarizer(article, max_length=int(max_len), min_length=int(min_len), do_sample=False)[0]["summary_text"]


iface = gr.Interface(fn=main,
                     inputs=[gr.inputs.Textbox(lines=5, label="Article text"),
                             "number",
                             "number"
                             ],
                     outputs="text")
iface.launch()