| import gradio as gr | |
| import joblib as jb | |
| def predict(sex, age, pclass): | |
| model = jb.load('model-teste.pkl') | |
| pclass = int(pclass) | |
| p = model.predict({{sex, age, pclass}})[0] | |
| return {"Did not survive": p[0], "Survived": p[1]} | |
| demo = gr.Interface(fn=predict, | |
| inputs=[gr.Dropdown(choices=["male", "female"], type="index"), | |
| "number", | |
| gr.Dropdown(choices=["1", "2", "3"], type="value") | |
| ], | |
| outputs="label") | |
| demo.launch() |