1

app.py

@@ -28,7 +28,7 @@ import matplotlib.pyplot as pl
 from transformers import pipeline
 from dust3r.utils.viz_demo import convert_scene_output_to_glb
 import depth_pro
-# import spaces
+import spaces
 from huggingface_hub import hf_hub_download
 pl.ion()
 
@@ -80,7 +80,8 @@ def get_3D_model_from_scene(outdir, silent, scene, min_conf_thr=3, as_pointcloud
     return convert_scene_output_to_glb(outdir, rgbimg, pts3d, msk, focals, cams2world, as_pointcloud=as_pointcloud,
                                         transparent_cams=transparent_cams, cam_size=cam_size, show_cam=show_cam, silent=silent, save_name=save_name,
                                         cam_color=cam_color)
-
+                                        
+@spaces.GPU(duration=180)
 def generate_monocular_depth_maps(img_list, depth_prior_name):
     depth_list = []
     focallength_px_list = []
@@ -115,7 +116,7 @@ def generate_monocular_depth_maps(img_list, depth_prior_name):
           #np.savez_compressed(path_depthanything, depth=depth)  
     return depth_list, focallength_px_list
 
-# @spaces.GPU(duration=180)
+@spaces.GPU(duration=180)
 def local_get_reconstructed_scene(filelist, min_conf_thr, as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, depth_prior_name, **kw):
     depth_list, focallength_px_list = generate_monocular_depth_maps(filelist, depth_prior_name)
     imgs = load_images(filelist, depth_list, focallength_px_list, size=image_size, verbose=not silent,traj_format='custom', depth_prior_name=depth_prior_name)