Update app.py

app.py

@@ -209,6 +209,7 @@ def update_category_models(category):
         if cat["category"] == category:
             model_names = [model[0] for model in cat["models"]]
             if model_names:
+                # Return the list of choices first, then the default value
                 return model_names, model_names[0]
             return [], None
     return [], None
@@ -721,7 +722,7 @@ def create_app():
                             [model[0] for model in MODELS[0]["models"]],
                             label="Models in Category",
                             value=MODELS[0]["models"][0][0],
-                            allow_custom_value=True  # allow custom values
+                            allow_custom_value=True
                         )
                 
                 with gr.Accordion("Generation Parameters", open=False):
@@ -918,6 +919,7 @@ def create_app():
         model_categories.change(
             fn=update_category_models,
             inputs=model_categories,
+            # The first output updates the choices, the second updates the value
             outputs=[category_models, category_models]
         )