Update app.py

app.py

@@ -56,23 +56,6 @@ def install_cuda_toolkit():
     
 install_cuda_toolkit()
 
-import os
-import torch
-import numpy as np
-import trimesh
-from pytorch3d.io import load_obj
-
-from pixel3dmm import env_paths
-from pixel3dmm.tracking.flame.FLAME import FLAME
-from pixel3dmm.tracking.tracker import Tracker
-from omegaconf import OmegaConf
-
-
-DEVICE = "cuda"
-
-base_conf = OmegaConf.load(f'{env_paths.CODE_BASE}/configs/tracking.yaml')
-
-flame_model = FLAME(base_conf).to(DEVICE)
 
 
 # Utility to select first image from a folder
@@ -170,6 +153,24 @@ def step3_uv_map(state):
 # Step 4: Tracking → final tracking image
 @spaces.GPU()
 def step4_track(state):
+
+    import os
+    import torch
+    import numpy as np
+    import trimesh
+    from pytorch3d.io import load_obj
+    
+    from pixel3dmm.tracking.flame.FLAME import FLAME
+    from pixel3dmm.tracking.tracker import Tracker
+    from omegaconf import OmegaConf
+    
+    
+    DEVICE = "cuda"
+    
+    base_conf = OmegaConf.load(f'{env_paths.CODE_BASE}/configs/tracking.yaml')
+    
+    flame_model = FLAME(base_conf).to(DEVICE)
+
     session_id = state.get("session_id")
     base_conf.video_name = f'{session_id}'
     tracker = Tracker(base_conf, flame_model)