Update app.py

app.py

@@ -4,9 +4,9 @@ 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]
+    p = model.predict_proba([[sex, age, pclass]])[0]
 
-    return {"Did not survive": p[1], "Survived": p[0]}
+    return {"Did not survive": p[0], "Survived": p[1]}
 
 demo = gr.Interface(fn=predict,
                         inputs=[gr.Dropdown(choices=["male", "female"], type="index"),