Update app.py

app.py

@@ -152,53 +152,53 @@ def chat_interface(message, history, agent_cluster, temperature=0.9, max_new_tok
         active_agents = [agent for agent, is_active in agent_cluster.items() if is_active]
         response = generate(message, history, active_agents, temperature, max_new_tokens, top_p, repetition_penalty)
         return (message, response)
-
 with gr.Blocks(theme='ParityError/Interstellar') as demo: 
-    with gr.Row(): for agent_name, agent_data in agent_roles.items(): 
-    gr.Button(agent_name, variant="secondary").click(toggle_agent, inputs=[gr.Button], outputs=[gr.Textbox]) gr.Textbox(agent_data["description"], interactive=False)
-
-with gr.Row(:
-    gr.ChatInterface(
-        chat_interface,
-        additional_inputs=[
-            gr.Slider(
-                label="Temperature",
-                value=0.9,
-                minimum=0.0,
-                maximum=1.0,
-                step=0.05,
-                interactive=True,
-                info="Higher values generate more diverse outputs",
-            ),
-            gr.Slider(
-                label="Maximum New Tokens",
-                value=2048,
-                minimum=64,
-                maximum=4096,
-                step=64,
-                interactive=True,
-                info="The maximum number of new tokens",
-            ),
-            gr.Slider(
-                label="Top-p (Nucleus Sampling)",
-                value=0.90,
-                minimum=0.0,
-                maximum=1,
-                step=0.05,
-                interactive=True,
-                info="Higher values sample more low-probability tokens",
-            ),
-            gr.Slider(
-                label="Repetition Penalty",
-                value=1.2,
-                minimum=1.0,
-                maximum=2.0,
-                step=0.05,
-                interactive=True,
-                info="Penalize repeated tokens",
-            )
-        ],
-        inputs=[gr.Textbox, gr.Chatbot, get_agent_cluster],
-    )
-
+    with gr.Row():
+        for agent_name, agent_data in agent_roles.items(): 
+            gr.Button(agent_name, variant="secondary").click(toggle_agent, inputs=[gr.Button], outputs=[gr.Textbox]) 
+            gr.Textbox(agent_data["description"], interactive=False)
+
+    with gr.Row():
+        gr.ChatInterface(
+            chat_interface,
+            additional_inputs=[
+                gr.Slider(
+                    label="Temperature",
+                    value=0.9,
+                    minimum=0.0,
+                    maximum=1.0,
+                    step=0.05,
+                    interactive=True,
+                    info="Higher values generate more diverse outputs",
+                ),
+                gr.Slider(
+                    label="Maximum New Tokens",
+                    value=2048,
+                    minimum=64,
+                    maximum=4096,
+                    step=64,
+                    interactive=True,
+                    info="The maximum number of new tokens",
+                ),
+                gr.Slider(
+                    label="Top-p (Nucleus Sampling)",
+                    value=0.90,
+                    minimum=0.0,
+                    maximum=1,
+                    step=0.05,
+                    interactive=True,
+                    info="Higher values sample more low-probability tokens",
+                ),
+                gr.Slider(
+                    label="Repetition Penalty",
+                    value=1.2,
+                    minimum=1.0,
+                    maximum=2.0,
+                    step=0.05,
+                    interactive=True,
+                    info="Penalize repeated tokens",
+                )
+            ],
+            inputs=[gr.Textbox, gr.Chatbot, get_agent_cluster],
+        )
 demo.queue().launch(debug=True)