update?

app.py

@@ -17,24 +17,31 @@ model_dir = model.download()
 model = joblib.load(model_dir + "/titanic_model.pkl")
 
 df = pd
-features = pd.read_csv(
-    "https://raw.githubusercontent.com/Nathanotal/remoteFiles/main/titanicCleaned.csv")
-features = features.drop(columns=["survived"])
-featureLabels = features.columns
+# features = pd.read_csv(
+#     "https://raw.githubusercontent.com/Nathanotal/remoteFiles/main/titanicCleaned.csv")
+# features = features.drop(columns=["survived"])
+# featureLabels = features.columns
+featureLabels = ["Pclass", "Name", "Sex", "Age", "SibSp",
+                 "Parch", "Ticket", "Fare", "Cabin", "Embarked"]
 
 
-def titanic(input1, input2, input3, input4, input5, input6, input7, input8):
+def titanic(Pclass, Sex, Age, SibSp, Parch, Fare, Cabin, Embarked):
     input_list = []
 
-    # Add all inputs to the input_list
-    input_list.append(input1)
-    input_list.append(input2)
-    input_list.append(input3)
-    input_list.append(input4)
-    input_list.append(input5)
-    input_list.append(input6)
-    input_list.append(input7)
-    input_list.append(input8)
+    sexToFeature = {
+        "male": 0,
+        "female": 1,
+    }
+
+    # Convert inputs to features
+    input_list.append(Pclass)  # Todo: Convert to feature
+    input_list.append(sexToFeature.get(Sex))  # Todo: Convert to feature
+    input_list.append(Age)  # !
+    input_list.append(SibSp)  # Todo: Convert to feature
+    input_list.append(Parch)  # Todo: Convert to feature
+    input_list.append(Fare)  # Todo: Convert to feature
+    input_list.append(Cabin)  # Todo: Convert to feature
+    input_list.append(Embarked)  # Todo: Convert to feature
 
     # 'res' is a list of predictions returned as the label.
     res = model.predict(np.asarray(input_list).reshape(1, -1))
@@ -64,8 +71,21 @@ def titanic(input1, input2, input3, input4, input5, input6, input7, input8):
 
 
 inputs = []
+numericalInputs = ["age", "sibsp", "parch", "fare", "pclass"]
+worthlessInputs = ["name", "ticket"]
+categoricalInputs = ["sex", "embarked", "cabin"]
+
 for feature in featureLabels:
-    inputs.append(gr.inputs.Number(default=1.0, label=feature))
+    if feature in numericalInputs:
+        inputs.append(gr.inputs.Number(default=1.0, label=feature))
+    elif feature in worthlessInputs:
+        pass
+        # inputs.append(gr.Inputs.Textbox(default='text', label=feature))
+    elif feature in categoricalInputs:
+        inputs.append(gr.inputs.Dropdown(
+            choices=["a", "b"], default="a", label=feature))
+    else:
+        raise Exception(f'Feature: "{feature}" not found')
 
 demo = gr.Interface(
     fn=titanic,