app_01

app.py

@@ -0,0 +1,16 @@
+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()