Update app.py

app.py

@@ -1,4 +1,8 @@
 import spaces
+
+import torch._dynamo
+torch._dynamo.disable()
+
 import os
 import subprocess
 import tempfile
@@ -30,7 +34,6 @@ 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
 
 sh("cd src/pixel3dmm/preprocessing/facer && pip install -e . && cd ../../../..")
@@ -201,11 +204,8 @@ 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)
-
-    import torch._dynamo
-    
-    with torch._dynamo.disable():
-        tracker.run()
+    tracker.run()
+        
 
     tracking_dir = os.path.join(os.environ["PIXEL3DMM_TRACKING_OUTPUT"], session_id, "frames")
     image = first_image_from_dir(tracking_dir)