gsasikiran's picture
BUGFIX: Return processed text
f72f135
raw
history blame contribute delete
661 Bytes
from transformers import pipeline
import gradio as gr
model = pipeline("summarization", model="gsasikiran/bart-base-finetuned-cnn", tokenizer="facebook/bart-base")
title = "Article summarizer"
description = "Generate summaries for articles based on the input text."
article = ""
def generate_summary(text):
summary_text = model(text)[0]['summary_text']
summary_text = summary_text.replace("\n", " ")
return summary_text
hf = gr.Interface(
fn=generate_summary,
inputs=gr.Textbox(lines=5, label="Input Text"),
title=title,
description=description,
article=article,
outputs = gr.Textbox(lines=2, label="Title")
)
hf.launch()