mv dpt pack to cpu after run_dpt

app.py

@@ -275,6 +275,10 @@ def run_depth_estimation(
             "scale": scale,
         }
     )
+
+    for k, v in dpt_pack.items():
+        if isinstance(v, torch.Tensor):
+            dpt_pack[k] = v.to('cpu')
     
     trimesh.PointCloud(scene_est_depth_pts.reshape(-1, 3), scene_est_depth_pts_colors.reshape(-1, 3)).\
         apply_translation(-trans).apply_scale(1. / (scale + 1e-6)).\