rebuild

app.py

@@ -294,17 +294,18 @@ with gr.Blocks(css=css, fill_height=True) as demo:
     with gr.Row(min_height=500, equal_height=True):
         # Left side: chat area
         with gr.Column(scale=7):
-            chat_interface = gr.ChatInterface(
-                fn=generate,
-                title="Conditionally Steer AI Responses Based on Topics",
-                description="""This is an experimental chatbot that you can steer using topics you care about:
+            gr.Markdown("""# Conditionally Steer AI Responses Based on Topics""")
+            gr.Markdown("""This is an experimental chatbot that you can steer using topics you care about:
 
 Step 1: Choose a topic to detect (e.g., "Google")      
 Step 2: Choose a topic you want the model to discuss when the previous topic comes up (e.g., "ethics")
 
-Try it out! For example, set it to detect "Google" topics and steer toward discussing "ethics", and ask "List some search engines and their pros and cons". We intervene on Gemma-2-2B-it by adding steering vectors to the residual stream at layer 20.""",
+We intervene on Gemma-2-2B-it by adding steering vectors to the residual stream at layer 20.""")
+            chat_interface = gr.ChatInterface(
+                fn=generate,
+                chatbot=gr.Chatbot(),
+                textbox=gr.Textbox(placeholder="List some search engines with their pros and cons", container=False, scale=7),
                 additional_inputs=[selected_detection, selected_subspaces],
-                fill_height=True,
             )
         
         # Right side: concept detection and steering