alexnasa commited on
Commit
fb2ed30
·
verified ·
1 Parent(s): 6ffadd1

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
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