zoedepth gpu dec

demo/mesh_recon.py

@@ -4,6 +4,9 @@ import numpy as np
 import torch
 import trimesh
 
+import spaces
+
+
 device = torch.device('cuda') if torch.cuda.is_available() else torch.device('cpu')
 
 # use torch hub
@@ -96,6 +99,7 @@ def depth_edges_mask(depth):
     return mask
 
 
+@spaces.GPU
 def mesh_reconstruction(
         masked_image: np.ndarray,
         mask: np.ndarray,