demo2 / app.py
amasiukevich's picture
Resolved the configuration of spaces
bf0ca6d
raw
history blame contribute delete
335 Bytes
from transformers import pipeline
import gradio as gr
model = pipeline("summarization")
def predict(prompt):
summary = model(prompt)[0]['summary_text']
return summary
textbox = gr.Textbox(placeholder="Enter text block to summarize", lines=4)
demo = gr.Interface(fn=predict, inputs=textbox, outputs="text")
demo.launch()