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()