xsum-bart-demo / app.py
mohadrk's picture
corregimos nombre del modelo
7b00e2f verified
raw
history blame contribute delete
399 Bytes
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()