variable name fixed

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

@@ -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(f'{env_paths.FLAME_ASSETS}', 'rb') as f:
+        with open(f'{env_paths.FLAME_ASSET}', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             flame_model = Struct(**ss)
 

src/pixel3dmm/tracking/flame/FLAME.py

@@ -69,7 +69,7 @@ class FLAME(nn.Module):
     def __init__(self, config):
         super(FLAME, self).__init__()
 
-        with open(f'{env_paths.FLAME_ASSETS}', 'rb') as f:
+        with open(f'{env_paths.FLAME_ASSET}', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             flame_model = Struct(**ss)
 

src/pixel3dmm/tracking/tracker.py

@@ -205,7 +205,7 @@ class Tracker(object):
         self.R_base = {}
         self.t_base = {}
 
-        flame_mesh_mask = np.load(f'{env_paths.FLAME_ASSETS}/FLAME2020/FLAME_masks/FLAME_masks.pkl', allow_pickle=True, encoding='latin1')
+        flame_mesh_mask = np.load(f'{env_paths.FLAME_ASSET}/FLAME2020/FLAME_masks/FLAME_masks.pkl', allow_pickle=True, encoding='latin1')
         self.vertex_face_mask = torch.from_numpy(flame_mesh_mask['face']).cuda().long()
 
 

src/pixel3dmm/utils/masking.py

@@ -48,14 +48,14 @@ class Struct(object):
 class Masking(nn.Module):
     def __init__(self):
 
-        dir = f'{env_paths.FLAME_ASSETS}'
+        dir = f'{env_paths.FLAME_ASSET}'
         super(Masking, self).__init__()
         with open(f'{dir}/FLAME2020/FLAME_masks/FLAME_masks.pkl', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             self.masks = Struct(**ss)
 
 
-        with open(f'{env_paths.FLAME_ASSETS}', 'rb') as f:
+        with open(f'{env_paths.FLAME_ASSET}', 'rb') as f:
             ss = pickle.load(f, encoding='latin1')
             flame_model = Struct(**ss)