Update src/pixel3dmm/tracking/flame/FLAME.py

src/pixel3dmm/tracking/flame/FLAME.py

@@ -66,7 +66,14 @@ class FLAME(nn.Module):
 
     def __init__(self, config):
         super(FLAME, self).__init__()
-        with open(f'{env_paths.FLAME_ASSETS}/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)