Spaces:
Sleeping
Sleeping
File size: 535 Bytes
0861518 5c6d46e 0861518 5c6d46e 0861518 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 |
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() |