File size: 399 Bytes
06a9d8c
 
 
7b00e2f
06a9d8c
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
import gradio as gr
from transformers import pipeline

summarizer = pipeline("summarization", model="mohadrk/bart-xsum-summarization")

def summarize_text(text):
    summary = summarizer(text, max_length=128, min_length=30, do_sample=False)
    return summary[0]['summary_text']

demo = gr.Interface(fn=summarize_text, inputs="textbox", outputs="textbox", title="BART Summarization")

demo.launch()