Update src/pixel3dmm/env_paths.py

src/pixel3dmm/env_paths.py

@@ -16,26 +16,26 @@ with env.prefixed("PIXEL3DMM_"):
 
 flame_local_path = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
-            filename="FLAME/generic_model.pkl",
+            filename="generic_model.pkl",
             repo_type="model", 
         )
 
 pipnet_local_path = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
-            filename="PIPNet/epoch59.pth",
+            filename="epoch59.pth",
             repo_type="model", 
         )
 
 normals_model_local_path = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
-            filename="FLAME/normals.ckpt",
-            repo_type="pretrained_weights", 
+            filename="normals.ckpt",
+            repo_type="model", 
         )
 
 uv_map_local_path = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
-            filename="FLAME/uv.ckpt",
-            repo_type="pretrained_weights", 
+            filename="uv.ckpt",
+            repo_type="model", 
         )
 
 head_template = f'{CODE_BASE}/assets/head_template.obj'