titanic v1

app.py

@@ -1,11 +1,17 @@
+from secrets import choice
 import gradio as gr
 import joblib as jb
 
-def predict(age, sex, bmi):
+def predict(sex, age, pclass):
     model = jb.load('model.pkl')
-    return model.predict([[age, sex, bmi]])
+    pclass += 1
+    p = model.predict_proba([[sex, age, pclass]])[0]
+
+    return {"Did not survive": p[0], "Survived": p[1]}
 
 demo = gr.Interface(fn=predict,
-                    inputs=["number", "number", "number"],
-                    outputs="number")
+                    inputs=[gr.Dropdown(choices=["male", "female"], type="index")
+                            , "number", 
+                            gr.Dropdown(choices=["1", "2", "3"], type="index")],
+                    outputs="label")
 demo.launch()

model.pkl

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:49bd09efbc136a8d7d4d77ee6169600c9538f308d0b6b4ab9d89d423aba1a718
-size 875
+oid sha256:4aa9451cefc9160e3bbd6b2d9ae47939b67d553d743a521ffb53c19930f945cb
+size 10591