Update src/pixel3dmm/preprocessing/MICA/models/flame.py

src/pixel3dmm/preprocessing/MICA/models/flame.py

@@ -21,7 +21,7 @@ import loguru
 import numpy as np
 import torch
 import torch.nn as nn
-
+from pixel3dmm import env_paths
 from .lbs import lbs, batch_rodrigues, vertices2landmarks, rot_mat_to_euler
 
 
@@ -52,7 +52,7 @@ class FLAME(nn.Module):
     def __init__(self, config, optimize_basis=False):
         super(FLAME, self).__init__()
         loguru.logger.info("[FLAME] creating the FLAME Decoder")
-        with open(config.flame_model_path, 'rb') as f:
+        with open(f'{env_paths.FLAME_ASSETS}', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             flame_model = Struct(**ss)