autosteer

app.py

@@ -241,18 +241,18 @@ with gr.Blocks(css=css, fill_height=True) as demo:
     
     with gr.Row(min_height=700):
         # Left side: bigger chat area
-        with gr.Column(scale=6):
+        with gr.Column(scale=7):
             chat_interface = gr.ChatInterface(
                 fn=generate,
                 title="Chat with a Concept Steering Model",
-                description="""Steer responses by selecting concepts on the right →\n\nWe are using Gemma-2-2B-it with steering vectors added to the residual stream at layer 20. Our auto-steer steering vector generated is a finetuned Gemma-2-2B model.""",
+                description="""Steer responses by selecting concepts on the right →\n\nWe are using Gemma-2-2B-it with steering vectors added to the residual stream at layer 20. Our auto-steer steering vector generated is a finetuned Gemma-2-2B model. Steered model tends to generate longer responses.""",
                 type="messages",
                 additional_inputs=[selected_subspaces],
                 fill_height=True
             )
         
         # Right side: concept management
-        with gr.Column(scale=4):
+        with gr.Column(scale=3):
             gr.Markdown("## Steer Model Responses")
             gr.Markdown("Search and then select a concept to steer. The closest match will be automatically selected. If there is no match, we will use our steering vector generator to auto-steer for you!")
             # Concept Search and Selection
@@ -266,7 +266,7 @@ with gr.Blocks(css=css, fill_height=True) as demo:
                 concept_dropdown = gr.Dropdown(
                     label="Select a concept to steer the model (Click to see more!)",
                     interactive=True,
-                    allow_custom_value=False
+                    allow_custom_value=False,
                 )
                 concept_magnitude = gr.Slider(
                     label="Steering Intensity",