Update src/pixel3dmm/utils/masking.py

src/pixel3dmm/utils/masking.py

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