Update app.py

app.py

@@ -502,6 +502,7 @@ def compare_groups(data_choice, model1, model2):
 
   return fig, "\n".join(p_values_text)
 import gradio as gr
+cats1 = list(DATASET_CATEGORIES.keys())
 models = ["RandomForest", "DecisionTree", "KNN", "SVM", "LogisticRegression"]
 with gr.Blocks() as demo:
   with gr.Tabs() as tabs:
@@ -529,15 +530,15 @@ with gr.Blocks() as demo:
         with gr.Row():
           gr.Markdown("# Comparing models")
           model_choice = gr.Dropdown(models, label = "Model 1", multiselect = True)
-          med = gr.Dropdown(DATASET_CATEGORIES[cats[0]].values(), label = cats[0],multiselect=True)
-          game = gr.Dropdown(DATASET_CATEGORIES[cats[1]].values(), label = cats[1],multiselect=True)
-          ed = gr.Dropdown(DATASET_CATEGORIES[cats[2]].values(), label = cats[2],multiselect=True)
+          med = gr.Dropdown(DATASET_CATEGORIES[cats1[0]].values(), label = cats1[0],multiselect=True)
+          game = gr.Dropdown(DATASET_CATEGORIES[cats1[1]].values(), label = cats1[1],multiselect=True)
+          ed = gr.Dropdown(DATASET_CATEGORIES[cats1[2]].values(), label = cats1[2],multiselect=True)
         with gr.Row():
-          bank = gr.Dropdown(DATASET_CATEGORIES[cats[3]].values(), label = cats[3],multiselect=True)
-          sci = gr.Dropdown(DATASET_CATEGORIES[cats[4]].values(), label = cats[4],multiselect=True)
-          social = gr.Dropdown(DATASET_CATEGORIES[cats[5]].values(), label = cats[5],multiselect=True)
-          ml = gr.Dropdown(DATASET_CATEGORIES[cats[6]].values(), label = cats[6],multiselect=True)
-          other = gr.Dropdown(DATASET_CATEGORIES[cats[7]].values(), label = cats[7],multiselect=True)
+          bank = gr.Dropdown(DATASET_CATEGORIES[cats1[3]].values(), label = cats1[3],multiselect=True)
+          sci = gr.Dropdown(DATASET_CATEGORIES[cats1[4]].values(), label = cats1[4],multiselect=True)
+          social = gr.Dropdown(DATASET_CATEGORIES[cats1[5]].values(), label = cats1[5],multiselect=True)
+          ml = gr.Dropdown(DATASET_CATEGORIES[cats1[6]].values(), label = cats1[6],multiselect=True)
+          other = gr.Dropdown(DATASET_CATEGORIES[cats1[7]].values(), label = cats1[7],multiselect=True)
           compare_button = gr.Button("Compare")
         with gr.Row():
           #output_plot = gr.Gallery(label="Results")