Update app.py

app.py

@@ -14,7 +14,7 @@ model = mr.get_model("titanic_modal_more_specs_grad_boosted", version=1)
 model_dir = model.download()
 model = joblib.load(model_dir + "/titanic_model.pkl")
 
-def titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,embarked_remapped,cabin_remapped):
+def titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,cabin):
     input_list = []
     input_list.append(pclass)
     input_list.append(sex)
@@ -23,12 +23,11 @@ def titanic(pclass,sex,age,sibsp,parch,embarked,fare_per_customer,embarked_remap
     input_list.append(parch)
     input_list.append(embarked)
     input_list.append(fare_per_customer)
-    input_list.append(embarked_remapped)
-    input_list.append(cabin_remapped)
+    input_list.append(cabin)
     # 'res' is a list of predictions returned as the label.
     #global res
     res = model.predict(np.asarray(input_list).reshape(1, -1))
-    return {('This guy will')}
+    return "This guy will"
     
 demo = gr.Interface(
     titanic,
@@ -36,13 +35,13 @@ demo = gr.Interface(
     description="Predict survivals. 0 for dead and 1 for survived. ",
     inputs=[
         gr.inputs.Number(default=1.0, label="pclass"),
-        gr.inputs.Number(default=1.0, label="gender(male=0, female=1)"),
+        gr.inputs.Number(default=1.0, label="gender"),
         gr.inputs.Number(default=1.0, label="age"),
         gr.inputs.Number(default=1.0, label="sibsp"),
         gr.inputs.Number(default=1.0, label="parch"),
-        gr.inputs.Number(default=1.0, label="embarked(C=1,S=2,Q=3)"),
+        gr.inputs.Number(default=1.0, label="embarked"),
         gr.inputs.Number(default=1.0, label="fare_per_customer"),
-        gr.inputs.Number(default=1.0, label="cabin(if the passanger has one cabin =1, else =0)"),
+        gr.inputs.Number(default=1.0, label="cabin"),
     ],
     outputs=gr.Textbox()
     )