Update app.py

app.py

@@ -40,6 +40,20 @@ def update_imgbox(choices):
     choices_plus = extend_choices(choices)
     return [gr.Image(None, label=m, visible=(m != 'NA')) for m in choices_plus]
 
+def choice_group_a(group_model_choice):
+    models=group_model_choice
+    load_fn(models)
+    model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
+    return (model_choice,output,current_models)
+
+def choice_group_b(group_model_choice):
+    output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]
+    return output
+
+def choice_group_c(group_model_choice):
+    current_models = [gr.Textbox(m, visible=False) for m in default_models]
+    return current_models   
+
 def gen_fn(model_str, prompt):
     if model_str == 'NA':
         return None
@@ -78,19 +92,15 @@ def make_me():
         with gr.Accordion("test", open=True):
             group_model_choice = gr.Dropdown(label="test Model", show_label=False, choices=models_test , allow_custom_value=True)
             group_model_choice.change(choice_group_a,group_model_choice,(model_choice,output,current_models))
+            group_model_choice.change(choice_group_b,group_model_choice,output)
+            group_model_choice.change(choice_group_c,group_model_choice,current_models)
         with gr.Row():
             gr.HTML("""
                 <div class="footer">
                 <p> Based on the <a href="https://huggingface.co/spaces/derwahnsinn/TestGen">TestGen</a> Space by derwahnsinn, the <a href="https://huggingface.co/spaces/RdnUser77/SpacIO_v1">SpacIO</a> Space by RdnUser77 and Omnibus's Maximum Multiplier!
                 </p>
             """)
-def choice_group_a(group_model_choice):
-    models=group_model_choice
-    load_fn(models)
-    model_choice = gr.CheckboxGroup(models, label=f' {num_models} different models selected', value=default_models, multiselect=True, max_choices=num_models, interactive=True, filterable=False)
-    output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]
-    current_models = [gr.Textbox(m, visible=False) for m in default_models]
-    return (model_choice,output,current_models)
+
 
 js_code = """