Update app.py

app.py

@@ -5,6 +5,7 @@ import requests
 
 import hopsworks
 import joblib
+import pandas as pd
 
 project = hopsworks.login()
 fs = project.get_feature_store()
@@ -37,8 +38,17 @@ def titanic(ticket_class, sex, port, fare, age, sibsp, parch):
     input_list.append(age)
     input_list.append(int(sibsp))
     input_list.append(int(parch))
+    data = {
+        "pclass": CLASS_TO_VALUE[ticket_class],
+        "sex": sex,
+        "embarked": PORT_TO_VALUE[port],
+        "age": age,
+        "sibsp": int(sibsp),
+        "parch": int(parch)
+    }
+    df = pd.DataFrame(data)
     # 'res' is a list of predictions returned as the label.
-    res = model.predict(np.asarray(input_list).reshape(1, -1))
+    res = model.predict(df)
     # We add '[0]' to the result of the transformed 'res', because 'res' is a list, and we only want
     # the first element.