Update app.py

app.py

@@ -30,9 +30,9 @@ sh("pip install -e .")
 # tell Python to re-scan site-packages now that the egg-link exists
 import importlib, site; site.addsitedir(site.getsitepackages()[0]); importlib.invalidate_caches()
 
-from pixel3dmm import env_paths
 import torch._dynamo
 torch._dynamo.disable()
+from pixel3dmm import env_paths
 
 sh("cd src/pixel3dmm/preprocessing/facer && pip install -e . && cd ../../../..")
 sh("cd src/pixel3dmm/preprocessing/PIPNet/FaceBoxesV2/utils && sh make.sh && cd ../../../../../..")
@@ -202,7 +202,11 @@ def step4_track(state):
     session_id = state.get("session_id")
     tracking_conf.video_name = f'{session_id}'
     tracker = Tracker(tracking_conf, flame_model, diff_renderer)
-    tracker.run()
+
+    from torch._dynamo import disable as no_dynamo
+    
+    with no_dynamo()
+        tracker.run()
 
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")
     image = first_image_from_dir(tracking_dir)