import pickle import gradio as gr def dialect_pred(text): loaded_model = pickle.load(open('./dialect_model.sav', 'rb')) return loaded_model.predict([text])[0] demo = gr.Interface(fn=dialect_pred, inputs="text",outputs= ["text"]) demo.launch()