Pointmap: ship-mode — preload all sizes, default to 1B

app.py

@@ -59,7 +59,7 @@ POINTMAP_MODELS = {
         "config": os.path.join(CONFIGS_DIR, "sapiens2_5b_pointmap_render_people-1024x768.py"),
     },
 }
-DEFAULT_SIZE = "0.4B"  # iteration mode
+DEFAULT_SIZE = "1B"
 
 FG_REPO = "facebook/sapiens-seg-foreground-1b-torchscript"
 FG_FILENAME = "sapiens_1b_seg_foreground_epoch_8_torchscript.pt2"
@@ -99,8 +99,9 @@ def _get_fg_model():
     return _fg_model
 
 
-print("[startup] pre-loading 0.4B (iteration mode) + fg/bg ...")
-_get_pointmap_model(DEFAULT_SIZE)
+print("[startup] pre-loading all pointmap sizes + fg/bg ...")
+for _size in POINTMAP_MODELS:
+    _get_pointmap_model(_size)
 _get_fg_model()
 print("[startup] ready.")
 
@@ -374,7 +375,7 @@ with gr.Blocks(title="Sapiens2 Pointmap", theme=gr.themes.Soft(), css=CUSTOM_CSS
     with gr.Row():
         with gr.Column(scale=2, min_width=320):
             size = gr.Radio(
-                choices=["0.4B"],  # iteration mode — re-add other sizes when shipping
+                choices=list(POINTMAP_MODELS.keys()),
                 value=DEFAULT_SIZE,
                 label="Model",
                 container=False,