Update app.py

app.py

@@ -42,11 +42,12 @@ def update_imgbox(choices):
 
 def choice_group_a(group_model_choice):
     for m in models_test:
+        print((group_model_choice,' ',m[0]))
         if group_model_choice==m[0]:
             models=m
     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)
+    return (model_choice,output,model_choice)
 
 def choice_group_b(group_model_choice):
     output = [gr.Image(label=m, min_width=170, height=170) for m in default_models]