Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
|
@@ -73,18 +73,8 @@ DEVICE = "cuda"
|
|
| 73 |
|
| 74 |
base_conf = OmegaConf.load(f'{env_paths.CODE_BASE}/configs/tracking.yaml')
|
| 75 |
|
| 76 |
-
_mesh_file = env_paths.head_template
|
| 77 |
flame_model = FLAME(base_conf).to(DEVICE)
|
| 78 |
|
| 79 |
-
_obj_faces = load_obj(_mesh_file)[1]
|
| 80 |
-
|
| 81 |
-
diff_renderer = NVDRenderer(
|
| 82 |
-
image_size=base_conf.size,
|
| 83 |
-
obj_filename=_mesh_file,
|
| 84 |
-
no_sh=False,
|
| 85 |
-
white_bg=True
|
| 86 |
-
).to(DEVICE)
|
| 87 |
-
|
| 88 |
|
| 89 |
# Utility to select first image from a folder
|
| 90 |
def first_image_from_dir(directory):
|
|
|
|
| 73 |
|
| 74 |
base_conf = OmegaConf.load(f'{env_paths.CODE_BASE}/configs/tracking.yaml')
|
| 75 |
|
|
|
|
| 76 |
flame_model = FLAME(base_conf).to(DEVICE)
|
| 77 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 78 |
|
| 79 |
# Utility to select first image from a folder
|
| 80 |
def first_image_from_dir(directory):
|