Update app.py

app.py

@@ -72,7 +72,7 @@ def fetch_models(task):
         return [model.modelId for model in models]
     except Exception as e:
         return [f"Error fetching models: {str(e)}"]
-
+task_choices=["text-generation", "image-classification", "text-classification", "translation"]
 # Gradio Interface
 with gr.Blocks() as demo:
     gr.Markdown("## Hugging Face Chatbot with Dynamic Model Selection")
@@ -81,11 +81,11 @@ with gr.Blocks() as demo:
         with gr.Column():
             # Task selection
             task_selector = gr.Dropdown(
-                choices=["text-generation", "image-classification", "text-classification", "translation"],
+                choices=task_choices,
                 label="Select Task",
                 value="text-generation"
             )
-
+            models=fetch_models(task_choices[0])
             # Model selector
             model_selector = gr.Dropdown(choices=[], label="Select Model")
             # Example of handling model updates