Update app.py

app.py

@@ -1,5 +1,5 @@
 import gradio as gr
-from huggingface_hub import InferenceClient, list_models
+from huggingface_hub import InferenceClient
 import os
 import json
 
@@ -50,33 +50,22 @@ def get_prompt(name):
     """Retrieve a system prompt by name."""
     return system_prompts.get(name, "")
 
-def fetch_models(task):
-    """Fetch models for a specific task from Hugging Face Hub."""
-    try:
-        all_models=list_models()
-        print(all_models)
-        models = list_models(filter=f"pipeline_tags:{task}")
-        return [model.modelId for model in models]
-    except Exception as e:
-        return [f"Error fetching models: {str(e)}"]
+# List of available models
+available_models = [
+    "gpt-3.5-turbo",
+    "gpt-4",
+    "HuggingFaceH4/zephyr-7b-beta",
+    "HuggingFaceH4/zephyr-7b-alpha"
+]
 
 # Gradio Interface
 with gr.Blocks() as demo:
-    gr.Markdown("## Hugging Face Chatbot with Dynamic Model Selection")
+    gr.Markdown("## Hugging Face Chatbot with Gradio")
 
     with gr.Row():
         with gr.Column():
-            # Task selection
-            task_selector = gr.Dropdown(
-                choices=["text-generation", "image-classification", "text-classification", "translation"],
-                label="Select Task",
-                value="text-generation"
-            )
-
-            # Model selector
-            model_selector = gr.Dropdown(choices=[], label="Select Model")
+            model_selector = gr.Dropdown(choices=available_models, label="Select Model", value=available_models[0])
 
-            # System prompt and input
             system_prompt_name = gr.Dropdown(choices=list(system_prompts.keys()), label="Select System Prompt")
             system_prompt_content = gr.TextArea(label="System Prompt", value=get_prompt("default"), lines=4)
             save_prompt_button = gr.Button("Save System Prompt")
@@ -87,14 +76,14 @@ with gr.Blocks() as demo:
         with gr.Column():
             output = gr.TextArea(label="Model Response", interactive=False, lines=10)
 
-    # Update model list when task changes
-    def update_model_list(task):
-        models = fetch_models(task)
-        print(f"Models:{models}")
-        return gr.Dropdown.update(choices=models, value=models[0] if models else None)
+    def load_prompt(name):
+        return get_prompt(name)
 
-    # Event bindings
-    task_selector.change(update_model_list, inputs=[task_selector], outputs=[model_selector])
+    system_prompt_name.change(
+        lambda name: (name, get_prompt(name)), 
+        inputs=[system_prompt_name], 
+        outputs=[system_prompt_name, system_prompt_content]
+    )
     save_prompt_button.click(update_prompt, inputs=[system_prompt_name, system_prompt_content], outputs=[])
     submit_button.click(chat_with_model, inputs=[user_input, system_prompt_content, model_selector], outputs=[output])