Pointmap: re-layout (Input ↔ 3D in row1, controls ↔ Pointmap heatmap in row2); rename label

app.py

@@ -395,27 +395,29 @@ HEADER_HTML = """
 with gr.Blocks(title="Sapiens2 Pointmap", theme=gr.themes.Soft(), css=CUSTOM_CSS) as demo:
     gr.HTML(HEADER_HTML)
 
-    with gr.Row():
+    # Row 1: input ↔ 3D mesh, equal height
+    with gr.Row(equal_height=True):
         inp = gr.Image(label="Input", type="pil", height=640, scale=2)
-        with gr.Column(scale=3):
-            out_glb = gr.Model3D(
-                label="3D mesh  ·  drag to orbit  ·  scroll to zoom  ·  shift+drag to pan",
-                height=640,
-                clear_color=[0.10, 0.11, 0.14, 1.0],
-                camera_position=(35, 70, 4.0),
-                zoom_speed=0.7,
-                pan_speed=0.5,
-            )
-            out_depth = gr.Image(label="Depth (Z)", type="pil", height=640)
-
-    with gr.Row():
-        size = gr.Radio(
-            choices=["0.4B"],  # iteration mode — re-add other sizes when shipping
-            value=DEFAULT_SIZE,
-            label="Model",
-            scale=4,
+        out_glb = gr.Model3D(
+            label="3D mesh  ·  drag to orbit  ·  scroll to zoom  ·  shift+drag to pan",
+            height=640,
+            clear_color=[0.10, 0.11, 0.14, 1.0],
+            camera_position=(35, 70, 4.0),
+            zoom_speed=0.7,
+            pan_speed=0.5,
+            scale=3,
         )
-        run = gr.Button("Run", variant="primary", size="lg", scale=1)
+
+    # Row 2: controls under the input, depth heatmap directly under the 3D mesh
+    with gr.Row(equal_height=True):
+        with gr.Column(scale=2):
+            size = gr.Radio(
+                choices=["0.4B"],  # iteration mode — re-add other sizes when shipping
+                value=DEFAULT_SIZE,
+                label="Model",
+            )
+            run = gr.Button("Run", variant="primary", size="lg")
+        out_depth = gr.Image(label="Pointmap", type="pil", height=640, scale=3)
 
     gr.Examples(examples=EXAMPLES, inputs=inp, examples_per_page=14)