Update app.py

app.py

@@ -13,7 +13,7 @@ from torchvision import transforms
 
 # ----------------- HELPER FUNCTIONS ----------------- #
 
-#ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")
+ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")
 
 LABELS_TO_IDS = {
     "Background": 0,
@@ -133,8 +133,8 @@ 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/depth/checkpoints/sapiens_0.3b/sapiens_0.3b_render_people_epoch_100_torchscript.pt2?download=true"
-#CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
-model_path = os.path.join(os.path.dirname(__file__), "sapiens_0.3b_goliath_best_goliath_mIoU_7673_epoch_194_torchscript.pt2")
+CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
+model_path = os.path.join(os.path.dirname(__file__), "sapiens_0.3b_render_people_epoch_100_torchscript.pt2")
 
 if not os.path.exists(model_path):
     os.makedirs(CHECKPOINTS_DIR, exist_ok=True)