Update app.py

app.py

@@ -142,6 +142,7 @@ def generate_mask(image: np.ndarray):
     mask_np = np.array(mask) / 255.0
     return mask_np
 
+@spaces.GPU
 @torch.no_grad()
 def reconstruct(video_path, conf_thresh, kf_every, as_pointcloud=False, remove_background=False):
     # Extract frames from video