Update app.py

app.py

@@ -14,6 +14,8 @@ from torchvision import transforms
 # ----------------- HELPER FUNCTIONS ----------------- #
 
 ASSETS_DIR = os.path.join(os.path.dirname(__file__), "assets")
+os.makedirs(ASSETS_DIR, exist_ok=True)
+
 
 LABELS_TO_IDS = {
     "Background": 0,
@@ -134,6 +136,8 @@ 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")
+os.makedirs(CHECKPOINTS_DIR, exist_ok=True)
+
 model_path = os.path.join(CHECKPOINTS_DIR, "sapiens_2b_normal_render_people_epoch_70_torchscript.pt2")
 
 if not os.path.exists(model_path):