Update app.py

app.py

@@ -180,47 +180,143 @@ def hbdscan_tranform(df_transformed):
 
 # Shared inputs
 ann_inputs = [
-    gr.Slider(18, 90, step=1, label="Age"),
-    gr.Dropdown(["Male", "Female"], label="Gender"),
-    gr.Dropdown(["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", "Local-gov", "State-gov", "Without-pay", "Never-worked"], label="Workclass"),
-    gr.Dropdown(["Preschool", "1st-4th", "5th-6th", "7th-8th", "9th", "10th", "11th", "12th", "HS-grad", "Some-college", "Assoc-voc", "Assoc-acdm", "Bachelors", "Masters", "Doctorate", "Prof-school"], label="Education"),
-    gr.Dropdown(["Married-civ-spouse", "Divorced", "Never-married", "Separated", "Widowed", "Married-spouse-absent", "Married-AF-spouse"], label="Marital Status"),
-    gr.Dropdown(["Tech-support", "Craft-repair", "Other-service", "Sales", "Exec-managerial", "Prof-specialty", "Handlers-cleaners", "Machine-op-inspct", "Adm-clerical", "Farming-fishing", "Transport-moving", "Priv-house-serv", "Protective-serv", "Armed-Forces"], label="Occupation"),
-    gr.Dropdown(["Wife", "Husband", "Own-child", "Not-in-family", "Other-relative", "Unmarried"], label="Relationship"),
-    gr.Dropdown(["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], label="Race"),
-    gr.Slider(0, 100000, step=100, label="Capital Gain"),
-    gr.Slider(0, 5000, step=50, label="Capital Loss"),
-    gr.Slider(1, 60, step=1, label="Hours Per Week"),
-    gr.Dropdown(["United-States", "Canada", "Mexico", "Other"], label="Native Country")
-]
+        gr.Slider(18, 90, step=1, label="Age"),
+        gr.Dropdown(
+            ["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", 
+             "Local-gov", "State-gov", "Without-pay", "Never-worked"], 
+            label="Workclass"
+        ),
+        gr.Dropdown(
+            ["Bachelors", "Some-college", "11th", "HS-grad", "Prof-school", 
+             "Assoc-acdm", "Assoc-voc", "9th", "7th-8th", "12th", "Masters", 
+             "1st-4th", "10th", "Doctorate", "5th-6th", "Preschool"], 
+            label="Education"
+        ),
+        gr.Dropdown(
+            ["Married-civ-spouse", "Divorced", "Never-married", "Separated", 
+             "Widowed", "Married-spouse-absent", "Married-AF-spouse"], 
+            label="Marital Status"
+        ),
+        gr.Dropdown(
+            ["Tech-support", "Craft-repair", "Other-service", "Sales", 
+             "Exec-managerial", "Prof-specialty", "Handlers-cleaners", 
+             "Machine-op-inspct", "Adm-clerical", "Farming-fishing", 
+             "Transport-moving", "Priv-house-serv", "Protective-serv", 
+             "Armed-Forces"], 
+            label="Occupation"
+        ),
+        gr.Dropdown(
+            ["Wife", "Husband", "Own-child", "Unmarried", "Other-relative", "Not-in-family"], 
+            label="Relationship"
+        ),
+        gr.Dropdown(
+            ["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], 
+            label="Race"
+        ),
+        gr.Dropdown(
+            ["Male", "Female"], 
+            label="Gender"
+        ),
+        gr.Slider(1, 60, step=1, label="Hours Per Week"),
+        gr.Slider(0, 100000, step=100, label="Capital Gain"),
+        gr.Slider(0, 5000, step=50, label="Capital Loss"),
+        gr.Dropdown(
+            ["United-States", "Other"], 
+            label="Native Country"
+        )
+    ]
 rf_inputs = [
-    gr.Slider(18, 90, step=1, label="Age"),
-    gr.Dropdown(["Male", "Female"], label="Gender"),
-    gr.Dropdown(["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", "Local-gov", "State-gov", "Without-pay", "Never-worked"], label="Workclass"),
-    gr.Dropdown(["Preschool", "1st-4th", "5th-6th", "7th-8th", "9th", "10th", "11th", "12th", "HS-grad", "Some-college", "Assoc-voc", "Assoc-acdm", "Bachelors", "Masters", "Doctorate", "Prof-school"], label="Education"),
-    gr.Dropdown(["Married-civ-spouse", "Divorced", "Never-married", "Separated", "Widowed", "Married-spouse-absent", "Married-AF-spouse"], label="Marital Status"),
-    gr.Dropdown(["Tech-support", "Craft-repair", "Other-service", "Sales", "Exec-managerial", "Prof-specialty", "Handlers-cleaners", "Machine-op-inspct", "Adm-clerical", "Farming-fishing", "Transport-moving", "Priv-house-serv", "Protective-serv", "Armed-Forces"], label="Occupation"),
-    gr.Dropdown(["Wife", "Husband", "Own-child", "Not-in-family", "Other-relative", "Unmarried"], label="Relationship"),
-    gr.Dropdown(["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], label="Race"),
-    gr.Slider(0, 100000, step=100, label="Capital Gain"),
-    gr.Slider(0, 5000, step=50, label="Capital Loss"),
-    gr.Slider(1, 60, step=1, label="Hours Per Week"),
-    gr.Dropdown(["United-States", "Canada", "Mexico", "Other"], label="Native Country")
-]
+        gr.Slider(18, 90, step=1, label="Age"),
+        gr.Dropdown(
+            ["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", 
+             "Local-gov", "State-gov", "Without-pay", "Never-worked"], 
+            label="Workclass"
+        ),
+        gr.Dropdown(
+            ["Bachelors", "Some-college", "11th", "HS-grad", "Prof-school", 
+             "Assoc-acdm", "Assoc-voc", "9th", "7th-8th", "12th", "Masters", 
+             "1st-4th", "10th", "Doctorate", "5th-6th", "Preschool"], 
+            label="Education"
+        ),
+        gr.Dropdown(
+            ["Married-civ-spouse", "Divorced", "Never-married", "Separated", 
+             "Widowed", "Married-spouse-absent", "Married-AF-spouse"], 
+            label="Marital Status"
+        ),
+        gr.Dropdown(
+            ["Tech-support", "Craft-repair", "Other-service", "Sales", 
+             "Exec-managerial", "Prof-specialty", "Handlers-cleaners", 
+             "Machine-op-inspct", "Adm-clerical", "Farming-fishing", 
+             "Transport-moving", "Priv-house-serv", "Protective-serv", 
+             "Armed-Forces"], 
+            label="Occupation"
+        ),
+        gr.Dropdown(
+            ["Wife", "Husband", "Own-child", "Unmarried", "Other-relative", "Not-in-family"], 
+            label="Relationship"
+        ),
+        gr.Dropdown(
+            ["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], 
+            label="Race"
+        ),
+        gr.Dropdown(
+            ["Male", "Female"], 
+            label="Gender"
+        ),
+        gr.Slider(1, 60, step=1, label="Hours Per Week"),
+        gr.Slider(0, 100000, step=100, label="Capital Gain"),
+        gr.Slider(0, 5000, step=50, label="Capital Loss"),
+        gr.Dropdown(
+            ["United-States", "Other"], 
+            label="Native Country"
+        )
+    ]
 hbd_inputs = [
-    gr.Slider(18, 90, step=1, label="Age"),
-    gr.Dropdown(["Male", "Female"], label="Gender"),
-    gr.Dropdown(["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", "Local-gov", "State-gov", "Without-pay", "Never-worked"], label="Workclass"),
-    gr.Dropdown(["Preschool", "1st-4th", "5th-6th", "7th-8th", "9th", "10th", "11th", "12th", "HS-grad", "Some-college", "Assoc-voc", "Assoc-acdm", "Bachelors", "Masters", "Doctorate", "Prof-school"], label="Education"),
-    gr.Dropdown(["Married-civ-spouse", "Divorced", "Never-married", "Separated", "Widowed", "Married-spouse-absent", "Married-AF-spouse"], label="Marital Status"),
-    gr.Dropdown(["Tech-support", "Craft-repair", "Other-service", "Sales", "Exec-managerial", "Prof-specialty", "Handlers-cleaners", "Machine-op-inspct", "Adm-clerical", "Farming-fishing", "Transport-moving", "Priv-house-serv", "Protective-serv", "Armed-Forces"], label="Occupation"),
-    gr.Dropdown(["Wife", "Husband", "Own-child", "Not-in-family", "Other-relative", "Unmarried"], label="Relationship"),
-    gr.Dropdown(["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], label="Race"),
-    gr.Slider(0, 100000, step=100, label="Capital Gain"),
-    gr.Slider(0, 5000, step=50, label="Capital Loss"),
-    gr.Slider(1, 60, step=1, label="Hours Per Week"),
-    gr.Dropdown(["United-States", "Canada", "Mexico", "Other"], label="Native Country")
-]
+        gr.Slider(18, 90, step=1, label="Age"),
+        gr.Dropdown(
+            ["Private", "Self-emp-not-inc", "Self-emp-inc", "Federal-gov", 
+             "Local-gov", "State-gov", "Without-pay", "Never-worked"], 
+            label="Workclass"
+        ),
+        gr.Dropdown(
+            ["Bachelors", "Some-college", "11th", "HS-grad", "Prof-school", 
+             "Assoc-acdm", "Assoc-voc", "9th", "7th-8th", "12th", "Masters", 
+             "1st-4th", "10th", "Doctorate", "5th-6th", "Preschool"], 
+            label="Education"
+        ),
+        gr.Dropdown(
+            ["Married-civ-spouse", "Divorced", "Never-married", "Separated", 
+             "Widowed", "Married-spouse-absent", "Married-AF-spouse"], 
+            label="Marital Status"
+        ),
+        gr.Dropdown(
+            ["Tech-support", "Craft-repair", "Other-service", "Sales", 
+             "Exec-managerial", "Prof-specialty", "Handlers-cleaners", 
+             "Machine-op-inspct", "Adm-clerical", "Farming-fishing", 
+             "Transport-moving", "Priv-house-serv", "Protective-serv", 
+             "Armed-Forces"], 
+            label="Occupation"
+        ),
+        gr.Dropdown(
+            ["Wife", "Husband", "Own-child", "Unmarried", "Other-relative", "Not-in-family"], 
+            label="Relationship"
+        ),
+        gr.Dropdown(
+            ["White", "Black", "Asian-Pac-Islander", "Amer-Indian-Eskimo", "Other"], 
+            label="Race"
+        ),
+        gr.Dropdown(
+            ["Male", "Female"], 
+            label="Gender"
+        ),
+        gr.Slider(1, 60, step=1, label="Hours Per Week"),
+        gr.Slider(0, 100000, step=100, label="Capital Gain"),
+        gr.Slider(0, 5000, step=50, label="Capital Loss"),
+        gr.Dropdown(
+            ["United-States", "Other"], 
+            label="Native Country"
+        )
+    ]
 
 # Interfaces for each model
 ann_interface = gr.Interface(