Pose: enable all 4 sizes; default = 1B (shipped)

app.py

@@ -63,7 +63,7 @@ POSE_MODELS = {
         "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_keypoints308_shutterstock_goliath_3po-1024x768.py"),
     },
 }
-DEFAULT_SIZE = "0.4B"  # iteration mode — only this is preloaded; others lazy-load on click
+DEFAULT_SIZE = "1B"
 
 DETECTOR_MODEL_ID = "facebook/detr-resnet-50"  # COCO person = label 1
 
@@ -110,13 +110,10 @@ def _get_pose_model(size: str):
     return _pose_model_cache[size]
 
 
-# Iteration mode: only preload the default (0.4B) for fast Space boot.
-# Re-enable full preload by uncommenting the loop below.
-print("[startup] pre-loading detector + 0.4B (iteration mode) ...")
+print("[startup] pre-loading detector + all pose sizes ...")
 _get_detector()
-_get_pose_model(DEFAULT_SIZE)
-# for _size in POSE_MODELS:
-#     _get_pose_model(_size)
+for _size in POSE_MODELS:
+    _get_pose_model(_size)
 print("[startup] ready.")
 
 
@@ -296,7 +293,7 @@ with gr.Blocks(title="Sapiens2 Pose", theme=gr.themes.Soft(), css=CUSTOM_CSS) as
 
     with gr.Row():
         size = gr.Radio(
-            choices=["0.4B"],  # iteration mode — re-add other sizes when shipping
+            choices=list(POSE_MODELS.keys()),
             value=DEFAULT_SIZE,
             label="Model",
             scale=2,