Pointmap: shrink Run button (drop size=lg) + container=False on Radio so they don't fill the column

app.py

@@ -456,16 +456,18 @@ with gr.Blocks(title="Sapiens2 Pointmap", theme=gr.themes.Soft(), css=CUSTOM_CSS
             scale=3,
         )
 
-    # Row 2: controls under the input, depth heatmap directly under the 3D mesh.
-    # NOT equal_height — otherwise the depth image (640px) stretches the Run button.
+    # Row 2: controls (left) | depth heatmap (right). The controls are wrapped
+    # in a fixed-height container with explicit min_width so the Run button
+    # doesn't stretch to match the 640px depth image.
     with gr.Row():
-        with gr.Column(scale=2):
+        with gr.Column(scale=2, min_width=320):
             size = gr.Radio(
                 choices=["0.4B"],  # iteration mode — re-add other sizes when shipping
                 value=DEFAULT_SIZE,
                 label="Model",
+                container=False,
             )
-            run = gr.Button("Run", variant="primary", size="lg")
+            run = gr.Button("Run", variant="primary")
         out_depth = gr.Image(label="Depth (Z)", type="pil", height=640, scale=3)
 
     gr.Examples(examples=EXAMPLES, inputs=inp, examples_per_page=14)