Spaces:
Running
on
Zero
Running
on
Zero
Update src/pixel3dmm/env_paths.py
Browse files
src/pixel3dmm/env_paths.py
CHANGED
@@ -14,7 +14,7 @@ with env.prefixed("PIXEL3DMM_"):
|
|
14 |
PREPROCESSED_DATA = env("PREPROCESSED_DATA")
|
15 |
TRACKING_OUTPUT = env("TRACKING_OUTPUT")
|
16 |
|
17 |
-
|
18 |
repo_id="alexnasa/pixel3dmm",
|
19 |
filename="generic_model.pkl",
|
20 |
repo_type="model",
|
@@ -49,7 +49,7 @@ EYE_MASK = f'{CODE_BASE}/assets/uv_mask_eyes.png'
|
|
49 |
FLAME_UV_COORDS = f'{CODE_BASE}/assets/flame_uv_coords.npy'
|
50 |
VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
|
51 |
VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
|
52 |
-
|
53 |
|
54 |
# paths to pretrained pixel3dmm checkpoints
|
55 |
CKPT_UV_PRED = f'{CODE_BASE}/pretrained_weights/uv.ckpt'
|
|
|
14 |
PREPROCESSED_DATA = env("PREPROCESSED_DATA")
|
15 |
TRACKING_OUTPUT = env("TRACKING_OUTPUT")
|
16 |
|
17 |
+
FLAME_ASSETS = hf_hub_download(
|
18 |
repo_id="alexnasa/pixel3dmm",
|
19 |
filename="generic_model.pkl",
|
20 |
repo_type="model",
|
|
|
49 |
FLAME_UV_COORDS = f'{CODE_BASE}/assets/flame_uv_coords.npy'
|
50 |
VALID_VERTS_NARROW = f'{CODE_BASE}/assets/uv_valid_verty_noEyes.npy'
|
51 |
VALID_VERTS = f'{CODE_BASE}/assets/uv_valid_verty_noEyes_noEyeRegion_debug_wEars.npy'
|
52 |
+
|
53 |
|
54 |
# paths to pretrained pixel3dmm checkpoints
|
55 |
CKPT_UV_PRED = f'{CODE_BASE}/pretrained_weights/uv.ckpt'
|