Update app.py

app.py

@@ -106,6 +106,7 @@ def step1_trim(video_path, seconds, fps, state):
     return f"✅ Step 1: Trimmed to {seconds}s @{fps}fps", state
 
 # Step 2: Preprocessing → cropped video
+@spaces.GPU()
 def step2_preprocess(state):
     session_id = state["session_id"]
     base_dir = state["base_dir"]
@@ -128,6 +129,7 @@ def step2_preprocess(state):
     return "✅ Step 2: Preprocessing complete", video, state
 
 # Step 3: Normals inference → normals video
+@spaces.GPU()
 def step3_normals(state):
     session_id = state["session_id"]
     base_dir = state["base_dir"]
@@ -149,6 +151,7 @@ def step3_normals(state):
     return "✅ Step 3: Normals inference complete", video, state
 
 # Step 4: UV map inference → uv map video
+@spaces.GPU()
 def step4_uv_map(state):
     session_id = state["session_id"]
     base_dir = state["base_dir"]