File size: 368 Bytes
6e02f1c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
from transformers import pipeline
import gradio as gr
model = pipeline('summarization')
def predict(prompt):
summary = model(prompt)[0]['summary_text']
return summary
with gr.Blocks() as demo:
textbox = gr.Textbox(placeholder = 'Enter text block to summarize', lines = 4)
gr.Interface(fn = predict, inputs = textbox, outputs = 'text')
demo.launch() |