Update src/pixel3dmm/env_paths.py

src/pixel3dmm/env_paths.py

@@ -27,19 +27,19 @@ MICA_TAR_ASSET = hf_hub_download(
         )
 
 
-pipnet_local_path = hf_hub_download(
+PIPNET_LOCAL_ASSET = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
             filename="epoch59.pth",
             repo_type="model", 
         )
 
-normals_model_local_path = hf_hub_download(
+CKPT_N_PRED = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
             filename="normals.ckpt",
             repo_type="model", 
         )
 
-uv_map_local_path = hf_hub_download(
+CKPT_UV_PRED = hf_hub_download(
             repo_id="alexnasa/pixel3dmm",
             filename="uv.ckpt",
             repo_type="model", 
@@ -58,6 +58,3 @@ VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
 VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
 
 
-# paths to pretrained pixel3dmm checkpoints
-CKPT_UV_PRED = f'{CODE_BASE}/pretrained_weights/uv.ckpt'
-CKPT_N_PRED = f'{CODE_BASE}/pretrained_weights/normals.ckpt'