Aalaa's picture
Update app.py
cdff003
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()