Update app.py

app.py

@@ -511,15 +511,20 @@ with gr.Blocks(css="""
                    )
                
                with gr.Column(visible=False) as hf_options:
-                   hf_model = gr.Dropdown(
-                       choices=list(MODEL_CONTEXT_SIZES["HuggingFace Inference"].keys()),
-                       value="microsoft/phi-3-mini-4k-instruct",
-                       label="HuggingFace Model"
-                   )
-                   hf_api_key = gr.Textbox(
-                       label="🔑 HuggingFace API Key",
-                       type="password"
-                   )
+                    hf_model = gr.Dropdown(
+                        choices=list(model_registry.hf_models.keys()),
+                        label="🔧 HuggingFace Model",
+                        value="Phi-3 Mini 4K"
+                    )
+                    hf_custom_model = gr.Textbox(  # This needs to be defined before being used
+                        label="Custom Model ID",
+                        placeholder="Enter custom model ID...",
+                        visible=False
+                    )
+                    hf_api_key = gr.Textbox(
+                        label="🔑 HuggingFace API Key",
+                        type="password"
+                    )
                
                with gr.Column(visible=False) as groq_options:
                    groq_model = gr.Dropdown(