File size: 334 Bytes
ffbadc4
471985d
ffbadc4
 
471985d
 
 
 
ffbadc4
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
import gradio as gr
import pickle

def greet(name):
    # transform query from user
    file = open('ms-marco-TinyBERT-L-2.sav','rb')
    model = pickle.load(file)
    file.close()
    return "Hello " + name + "!"

demo = gr.Interface(fn=greet, inputs="textbox", outputs="textbox")
    
if __name__ == "__main__":
    demo.launch()