Update app.py

app.py

@@ -209,9 +209,9 @@ def update_category_models(category):
         if cat["category"] == category:
             model_names = [model[0] for model in cat["models"]]
             if model_names:
-                return gr.Dropdown.update(choices=model_names, value=model_names[0])
-            return gr.Dropdown.update(choices=[], value=None)
-    return gr.Dropdown.update(choices=[], value=None)
+                return model_names, model_names[0]
+            return [], None
+    return [], None
 
 def encode_image_to_base64(image_path):
     """Encode an image file to base64 string"""
@@ -897,7 +897,7 @@ def create_app():
         model_search.change(
             fn=filter_models,
             inputs=model_search,
-            outputs=[model_choice, model_choice]  # update both values
+            outputs=[model_choice, model_choice]
         )
         
         # Update context display when model changes
@@ -918,12 +918,12 @@ def create_app():
         model_categories.change(
             fn=update_category_models,
             inputs=model_categories,
-            outputs=category_models
+            outputs=[category_models, category_models]
         )
         
         # Update main model choice when category model is selected
         category_models.change(
-            fn=lambda x: x,  # Simply pass the value through
+            fn=lambda x: x,
             inputs=category_models,
             outputs=model_choice
         )