o1 impl

app.py

@@ -188,9 +188,9 @@ with gr.Blocks(css="style.css") as demo:
         with gr.Column(scale=7):
             chat_interface = gr.ChatInterface(
                 fn=generate,
-                additional_inputs=[],  # we'll put the max tokens slider below
                 title="",
                 type="messages",
+                additional_inputs=[selected_subspaces],
             )
         # Right side: concept management
         with gr.Column(scale=3):
@@ -250,9 +250,4 @@ with gr.Blocks(css="style.css") as demo:
         [selected_subspaces, active_subspaces_table, remove_dropdown]
     )
 
-    # Link the slider back to chat generation
-    chat_interface.config(
-        extra_inputs=[max_token_slider, selected_subspaces]
-    )
-
     demo.launch()