Update app.py

app.py

@@ -46,17 +46,17 @@ def titanic(pclass, sex, age, fare, embarked, familysize, appellation, cabin):
 
     # Appellation
     if appellation == "master":
-        input_list.append(1).append(0).append(0).append(0).append(0).append(0)
+        input_list.append(1,0,0,0,0,0)
     elif appellation == "miss":
-        input_list.append(0).append(1).append(0).append(0).append(0).append(0)
+        input_list.append(0,1,0,0,0,0)
     elif appellation == "mr":
-        input_list.append(0).append(0).append(1).append(0).append(0).append(0)
+        input_list.append(0,0,1,0,0,0)
     elif appellation == "mrs":
-        input_list.append(0).append(0).append(0).append(1).append(0).append(0)
+        input_list.append(0,0,0,1,0,0)
     elif appellation == "officer":
-        input_list.append(0).append(0).append(0).append(0).append(1).append(0)
+        input_list.append(0,0,0,0,1,0)
     elif appellation == "royalty":
-        input_list.append(0).append(0).append(0).append(0).append(0).append(1)
+        input_list.append(0,0,0,0,0,1)
 
     # Cabin
     if cabin == "A":
@@ -82,7 +82,6 @@ def titanic(pclass, sex, age, fare, embarked, familysize, appellation, cabin):
     # 'res' is a list of predictions returned as the label.
     res = model.predict(np.asarray(input_list).reshape(1, -1))
     res = res.astype(int)
-    print(res)
     # We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want
     # the first element.
     titanic_url = "https://github.com/Qinglin2000/ID2223/blob/main/" + str(res[0]) + ".png?raw=true"
@@ -92,7 +91,7 @@ def titanic(pclass, sex, age, fare, embarked, familysize, appellation, cabin):
 
 demo = gr.Interface(
     fn=titanic,
-    title="Titatnic Predictive Analytics",
+    title="Titanic Predictive Analytics",
     description="Experiment with titanic dataset values.",
     allow_flagging="never",
     inputs=[
@@ -106,5 +105,4 @@ demo = gr.Interface(
         gr.Dropdown(choices=["A", "B", "C", "D", "E", "F", "G", "T", "U"], label="Cabin", value="A"),
     ],
     outputs=gr.Image(type="pil"))
-
 demo.launch()