fix additional models

app.py

@@ -239,20 +239,32 @@ def create_demo():
                 all_images = gr.State([])
                 all_results = gr.State([])
         
-       # Handle checkbox change to update model dropdown
+        # Handle checkbox change to update model dropdown
         def toggle_models(show_all, current_model):
+            # Make a fresh copy of the models lists to avoid any reference issues
+            preferred_choices = list(preferred_models)
+            all_choices = list(all_models_list)
+            
             if show_all:
-                return gr.Dropdown.update(choices=all_models, value=current_model)
+                # When showing all models, use the fresh copy of all models
+                return gr.Dropdown(choices=all_choices, value=current_model)
             else:
                 # Check if current model is in preferred models list
-                is_preferred = any(current_model == value for _, value in preferred_models)
+                preferred_values = [value for _, value in preferred_choices]
                 
-                if is_preferred:
+                if current_model in preferred_values:
                     # Keep the current model if it's in preferred models
-                    return gr.Dropdown.update(choices=preferred_models, value=current_model)
+                    return gr.Dropdown(choices=preferred_choices, value=current_model)
                 else:
                     # Reset to default model if current model is not in preferred models
-                    return gr.Dropdown.update(choices=preferred_models, value=default_model)
+                    return gr.Dropdown(choices=preferred_choices, value=default_model)
+        
+        # Connect checkbox to toggle model choices - keep this the same
+        show_all_models.change(
+            fn=toggle_models,
+            inputs=[show_all_models, model_choice],
+            outputs=[model_choice]
+        )
 
         # Update model info when model selection changes
         def update_model_info(model_value):
@@ -278,13 +290,6 @@ def create_demo():
             outputs=[model_choice]
         )
         
-        # Connect model selection to update info
-        model_choice.change(
-            fn=update_model_info,
-            inputs=[model_choice],
-            outputs=[model_info]
-        )
-        
         # Handle file uploads
         def handle_upload(files, current_paths, current_filenames):
             file_paths = []