Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -153,7 +153,7 @@ def step2_normals(session_id):
|
|
153 |
base_conf = OmegaConf.load("configs/base.yaml")
|
154 |
|
155 |
if "normals_model" not in _model_cache:
|
156 |
-
|
157 |
model = p3dmm_system.load_from_checkpoint(f"{env_paths.CKPT_N_PRED}", strict=False)
|
158 |
model = model.eval().to(DEVICE)
|
159 |
_model_cache["normals_model"] = model
|
|
|
153 |
base_conf = OmegaConf.load("configs/base.yaml")
|
154 |
|
155 |
if "normals_model" not in _model_cache:
|
156 |
+
print("----caching normal models----")
|
157 |
model = p3dmm_system.load_from_checkpoint(f"{env_paths.CKPT_N_PRED}", strict=False)
|
158 |
model = model.eval().to(DEVICE)
|
159 |
_model_cache["normals_model"] = model
|