pygmalion / app.py
Soumil30's picture
increased max words
9883bb6
import gradio as gr
from transformers import pipeline
summarizer = pipeline("summarization", model="facebook/bart-large-cnn")
def summarize(input):
return summarizer(input, max_length=250)[0]['summary_text']
iface = gr.Interface(fn=summarize, inputs="text", outputs="text")
iface.launch()