Update app.py

app.py

@@ -78,25 +78,33 @@ models = [
     #"meta-llama/Llama-3.3-70B-Instruct", #fail, need HF Pro subscription
 ]
 
-demo = gr.ChatInterface(
-    respond,
-    additional_inputs=[
-        gr.Textbox(value="", label="System message"),
-        gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max new tokens"),
-        gr.Slider(minimum=0.1, maximum=1.0, value=0.3, step=0.1, label="Temperature"),
-        gr.Slider(
-            minimum=0.1,
-            maximum=1.0,
-            value=0.95,
-            step=0.05,
-            label="Top-P",
-        ),
-        gr.Dropdown(choices=models, value=models[0], label="Model"),  # Dropdown for model selection
-    ],
-    fill_height=True,
-    chatbot=chatbot,
-    #theme="Nymbo/Nymbo_Theme",
-)
+# Add a title and move the model dropdown to the top
+with gr.Blocks() as demo:
+    gr.Markdown("# LLM Test (HF API)")  # Add a title to the top of the UI
+    
+    # Add the model dropdown above the chatbot
+    model_dropdown = gr.Dropdown(choices=models, value=models[0], label="Select Model")
+    
+    # Use the existing ChatInterface
+    gr.ChatInterface(
+        respond,
+        additional_inputs=[
+            gr.Textbox(value="", label="System message"),
+            gr.Slider(minimum=1, maximum=4096, value=1024, step=1, label="Max new tokens"),
+            gr.Slider(minimum=0.1, maximum=1.0, value=0.3, step=0.1, label="Temperature"),
+            gr.Slider(
+                minimum=0.1,
+                maximum=1.0,
+                value=0.95,
+                step=0.05,
+                label="Top-P",
+            ),
+            model_dropdown,  # Pass the dropdown as an additional input
+        ],
+        fill_height=True,
+        chatbot=chatbot,
+    )
+
 print("Gradio interface initialized.")
 
 if __name__ == "__main__":