File size: 524 Bytes
03dc223
 
65eccd2
03dc223
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
from transformers import pipeline
import gradio as gr
get_completion = pipeline("text-classification", model="facebook/bart-large-cnn")

def summarize(input):
    output = get_completion(input)
    return output[0]['summary_text']




gr.close_all()
demo = gr.Interface(fn=summarize,
                    inputs=[gr.Textbox(label="Text to summarize", lines=6)],
                    outputs=[gr.Textbox(label="Result", lines=3)],
                    title="Text summarization"

                   )
demo.launch(inline = False)