File size: 642 Bytes
85bd369
 
 
 
 
 
 
 
 
 
 
 
 
bfb0199
 
85bd369
 
 
 
 
 
 
 
 
bfb0199
85bd369
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
# 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()