Update app.py

app.py

@@ -26,6 +26,8 @@ def update_model_choices(provider):
     """Update model dropdown choices based on selected provider"""
     if provider == "Cohere":
         return gr.Dropdown(choices=COHERE_MODELS, value=COHERE_MODELS[0])
+    elif provider =="Cohere Labs":
+        return gr.Dropdown(choices=COHERE_LABS_MODELS, value=COHERE_LABS_MODELS[0])
     else:
         return gr.Dropdown(choices=[], value=None)
 
@@ -137,8 +139,8 @@ with gr.Blocks() as demo:
                         )
                         model = gr.Dropdown(
                             info="Model",
-                            choices=[COHERE_MODELS, COHERE_LABS_MODELS],
-                            value=[COHERE_MODELS[0], COHERE_LABS_MODELS[0]],
+                            choices=COHERE_MODELS,
+                            value=COHERE_MODELS[0],
                             elem_id="model_dropdown",
                             interactive=True,
                             show_label=False