Update app.py

app.py

@@ -172,16 +172,30 @@ def step4_track(state):
         from pytorch3d.io import load_obj
         
         from pixel3dmm.tracking.flame.FLAME import FLAME
+        from pixel3dmm.tracking.renderer_nvdiffrast import NVDRenderer
         from pixel3dmm.tracking.tracker import Tracker
     
         flame = FLAME(base_conf)      # CPU instantiation
         flame = flame.to(DEVICE)      # CUDA init happens here
         _model_cache["flame_model"] = flame   
+        
+        _mesh_file = env_paths.head_template
+        
+        _obj_faces = load_obj(_mesh_file)[1]
+        
+        _model_cache["diff_renderer"] = NVDRenderer(
+            image_size=base_conf.size, 
+            obj_filename=_mesh_file,
+            no_sh=False,
+            white_bg=True
+        ).to(DEVICE)
+
 
     flame_model = _model_cache["flame_model"]
+    diff_renderer = _model_cache["diff_renderer"]
     session_id = state.get("session_id")
     base_conf.video_name = f'{session_id}'
-    tracker = Tracker(base_conf, flame_model)
+    tracker = Tracker(base_conf, flame_model, diff_renderer)
     tracker.run()
 
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")