Update app.py

app.py

@@ -153,7 +153,6 @@ def step2_normals(session_id):
     base_conf = OmegaConf.load("configs/base.yaml")
 
     if "normals_model" not in _model_cache:
-        print("----caching normal models----")
         model = p3dmm_system.load_from_checkpoint(f"{env_paths.CKPT_N_PRED}", strict=False)
         model =  model.eval().to(DEVICE)
         _model_cache["normals_model"] = model   
@@ -205,6 +204,7 @@ def step4_track(session_id):
 
      # Lazy init + caching of FLAME model on GPU
     if "flame_model" not in _model_cache:
+        global _model_cache
 
         flame = FLAME(tracking_conf)      # CPU instantiation
         flame = flame.to(DEVICE)      # CUDA init happens here