Spaces:
Runtime error
Runtime error
import gradio as gr | |
import joblib as jb | |
def predict(sex, age, pclass): | |
model = jb.load('model.pkl') | |
pclass = int(pclass) | |
p = model.predict_proba([[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() |