Update app.py

app.py

@@ -55,7 +55,9 @@ def install_cuda_toolkit():
     print("==> finished installation")
     
 install_cuda_toolkit()
-    
+
+from omegaconf import OmegaConf
+
 DEVICE = "cuda"
 
 # 1. Prepare config at import time (no CUDA calls)
@@ -171,7 +173,6 @@ def step4_track(state):
         
         from pixel3dmm.tracking.flame.FLAME import FLAME
         from pixel3dmm.tracking.tracker import Tracker
-        from omegaconf import OmegaConf
     
         flame = FLAME(base_conf)      # CPU instantiation
         flame = flame.to(DEVICE)      # CUDA init happens here