Update app.py

app.py

@@ -124,11 +124,11 @@ def create_legend_image(labels_to_ids: dict[str, int], filename="legend.png"):
 
 # ----------------- MODEL ----------------- #
 
-URL = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/normal/checkpoints/sapiens_0.3b/sapiens_0.3b_normal_render_people_epoch_66_torchscript.pt2?download=true"
+URL = "https://huggingface.co/facebook/sapiens/resolve/main/sapiens_lite_host/torchscript/normal/checkpoints/sapiens_1b/sapiens_1b_normal_render_people_epoch_115_torchscript.pt2?download=true"
 CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
 os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
 
-model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_0.3b_normal_render_people_epoch_66_torchscript.pt2")
+model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_1b_normal_render_people_epoch_115_torchscript.pt2")
 
 if not os.path.exists(model_path) or os.path.getsize(model_path) == 0:
     print("Downloading model...")