app v1

app.py

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

requirements.txt

@@ -0,0 +1,3 @@
+joblib
+scikit-learn
+gradio