Update app.py

app.py

@@ -30,8 +30,7 @@ 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()
 
-import torch._dynamo
-torch._dynamo.disable()
+
 from pixel3dmm import env_paths
 
 sh("cd src/pixel3dmm/preprocessing/facer && pip install -e . && cd ../../../..")
@@ -203,9 +202,9 @@ def step4_track(state):
     tracking_conf.video_name = f'{session_id}'
     tracker = Tracker(tracking_conf, flame_model, diff_renderer)
 
-    from torch._dynamo import disable as no_dynamo
+    import torch._dynamo
     
-    with no_dynamo():
+    with torch._dynamo.disable():
         tracker.run()
 
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")