Update src/pixel3dmm/env_paths.py

src/pixel3dmm/env_paths.py

@@ -1,7 +1,7 @@
 import json
 from pathlib import Path
 from environs import Env
-
+from huggingface_hub import hf_hub_download
 
 env = Env(expand_vars=True)
 env_file_path = Path(f"{Path.home()}/.config/pixel3dmm/.env")
@@ -14,6 +14,29 @@ with env.prefixed("PIXEL3DMM_"):
     PREPROCESSED_DATA = env("PREPROCESSED_DATA")
     TRACKING_OUTPUT = env("TRACKING_OUTPUT")
 
+flame_local_path = hf_hub_download(
+            repo_id="alexnasa/pixel3dmm",
+            filename="FLAME/generic_model.pkl",
+            repo_type="model", 
+        )
+
+pipnet_local_path = hf_hub_download(
+            repo_id="alexnasa/pixel3dmm",
+            filename="PIPNet/epoch59.pth",
+            repo_type="model", 
+        )
+
+normals_model_local_path = hf_hub_download(
+            repo_id="alexnasa/pixel3dmm",
+            filename="FLAME/normals.ckpt",
+            repo_type="pretrained_weights", 
+        )
+
+uv_map_local_path = hf_hub_download(
+            repo_id="alexnasa/pixel3dmm",
+            filename="FLAME/uv.ckpt",
+            repo_type="pretrained_weights", 
+        )
 
 head_template = f'{CODE_BASE}/assets/head_template.obj'
 head_template_color = f'{CODE_BASE}/assets/head_template_color.obj'