Update app.py

app.py

@@ -134,7 +134,7 @@ def create_legend_image(labels_to_ids: dict[str, int], filename="legend.png"):
 
 URL = "https://huggingface.co/facebook/sapiens/blob/main/sapiens_lite_host/torchscript/normal/checkpoints/sapiens_2b/sapiens_2b_normal_render_people_epoch_70_torchscript.pt2?download=true"
 CHECKPOINTS_DIR = os.path.join(ASSETS_DIR, "checkpoints")
-model_path = os.path.join(os.path.dirname(__file__), "sapiens_2b_normal_render_people_epoch_70_torchscript.pt2")
+model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_2b_normal_render_people_epoch_70_torchscript.pt2")
 
 if not os.path.exists(model_path):
     os.makedirs(CHECKPOINTS_DIR, exist_ok=True)