Update src/pixel3dmm/preprocessing/MICA/utils/masking.py

src/pixel3dmm/preprocessing/MICA/utils/masking.py

@@ -15,6 +15,7 @@
 # Contact: [email protected]
 import os
 import pickle
+from huggingface_hub import hf_hub_download
 
 import numpy as np
 import torch
@@ -47,7 +48,14 @@ class Masking(nn.Module):
             ss = pickle.load(f, encoding='latin1')
             self.masks = Struct(**ss)
 
-        with open(f'{ROOT_DIR}/data/FLAME2020/generic_model.pkl', 'rb') as f:
+        
+        local_path = hf_hub_download(
+            repo_id="alexnasa/pixel3dmm",
+            filename="FLAME/generic_model.pkl",
+            repo_type="model", 
+        )
+     
+        with open(local_path, 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             flame_model = Struct(**ss)