Update app.py

app.py

@@ -50,29 +50,14 @@ 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:
         models = list_models(filter=f"pipeline_tags:{task}")
-        models += [
-            "HuggingFaceH4/zephyr-7b-beta",
-            "HuggingFaceH4/zephyr-7b-alpha",
-            "HuggingFaceH4/zephyr-6b"
-        ]
-
         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,16 +66,13 @@ with gr.Blocks() as demo:
         with gr.Column():
             # Task selection
             task_selector = gr.Dropdown(
-                choices=task_choices,
+                choices=["text-generation", "image-classification", "text-classification", "translation"],
                 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
-            model_selector.choices = models
-            model_selector.value = models[0]
 
             # System prompt and input
             system_prompt_name = gr.Dropdown(choices=list(system_prompts.keys()), label="Select System Prompt")
@@ -114,4 +96,4 @@ with gr.Blocks() as demo:
     submit_button.click(chat_with_model, inputs=[user_input, system_prompt_content, model_selector], outputs=[output])
 
 # Run the app
-demo.launch()
+demo.launch()