huggingface-git / app.py
Samson
dabbling
85bd369
raw
history blame
642 Bytes
# import gradio as gr
# def greet(name):
# return "Hello " + name + "!!... How are you doing today"
# iface = gr.Interface(fn=greet,
# inputs=gr.Textbox(lines=2, placeholder="Name Here..."),
# outputs="text")
# iface.launch()
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 what you need to summarize", lines = 4)
gr.Interface(fn=predict, input=textbox, output='text')
demo.launch()