Update app.py

app.py

@@ -50,6 +50,15 @@ def get_prompt(name):
     """Retrieve a system prompt by name."""
     return system_prompts.get(name, "")
 
+    
+def update_dropdown_choices(models):
+    """Update dropdown choices dynamically."""
+    if not models:
+        return [], None  # Return empty choices and no default value
+    return models, models[0]
+
+
+
 def fetch_models(task):
     """Fetch models for a specific task from Hugging Face Hub."""
     try:
@@ -79,6 +88,9 @@ with gr.Blocks() as demo:
 
             # Model selector
             model_selector = gr.Dropdown(choices=[], label="Select Model")
+            # Example of handling model updates
+            model_selector.choices = available_models
+            model_selector.value = available_models[0]
 
             # System prompt and input
             system_prompt_name = gr.Dropdown(choices=list(system_prompts.keys()), label="Select System Prompt")