init project

modules/pe3r/demo.py

@@ -77,7 +77,7 @@ def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world,
     scene.export(file_obj=outfile)
     return outfile
 
-
+@spaces.GPU(duration=180)
 def get_3D_model_from_scene(outdir, silent, scene, min_conf_thr=3, as_pointcloud=False, mask_sky=False,
                             clean_depth=False, transparent_cams=False, cam_size=0.05):
     """
@@ -424,7 +424,7 @@ def get_cog_feats(images, pe3r, device):
 
     return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
 
-
+@spaces.GPU(duration=180)
 def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, niter, min_conf_thr,
                             as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size,
                             scenegraph_type, winsize, refid):
@@ -511,6 +511,7 @@ def get_reconstructed_scene(outdir, pe3r, device, silent, filelist, schedule, ni
 
     return scene, outfile, imgs
 
+@spaces.GPU(duration=180)
 def get_3D_object_from_scene(outdir, pe3r, silent, device, text, threshold, scene, min_conf_thr, as_pointcloud, 
                  mask_sky, clean_depth, transparent_cams, cam_size):
     
@@ -547,7 +548,6 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
                               maximum=num_files - 1, step=1, visible=False)
     return winsize, refid
 
-@spaces.GPU(duration=180)
 def main_demo(tmpdirname, server_name, server_port, silent=False):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'