init project

app.py

@@ -447,7 +447,7 @@ def get_3D_model_from_scene(outdir, scene, min_conf_thr=3, as_pointcloud=False,
 #     return cog_seg_maps, rev_cog_seg_maps, multi_view_clip_feats
 
 
-@spaces.GPU(duration=30)
+@spaces.GPU#(duration=30)
 def get_reconstructed_scene(outdir, filelist, schedule='linear', niter=300, min_conf_thr=3.0,
                             as_pointcloud=True, mask_sky=False, clean_depth=True, transparent_cams=True, cam_size=0.05,
                             scenegraph_type='complete', winsize=1, refid=0):
@@ -537,11 +537,15 @@ def get_reconstructed_scene(outdir, filelist, schedule='linear', niter=300, min_
     outfile = get_3D_model_from_scene(outdir, scene, min_conf_thr, as_pointcloud, mask_sky,
                                       clean_depth, transparent_cams, cam_size)
     
-    scene.to('cpu')
+    # scene.to('cpu')
+    # print(scene)
+    # print(scene.imgs)
+    # print(scene.cogs) scene, 
+
     torch.cuda.empty_cache()
-    return scene, outfile 
+    return outfile 
 
-# @spaces.GPU(duration=30)
+# @spaces.GPU #(duration=30)
 # def get_3D_object_from_scene(outdir, text, threshold, scene, min_conf_thr, as_pointcloud, 
 #                  mask_sky, clean_depth, transparent_cams, cam_size):
     
@@ -569,7 +573,8 @@ recon_fun = functools.partial(get_reconstructed_scene, tmpdirname)
 
 with gradio.Blocks(css=""".gradio-container {margin: 0 !important; min-width: 100%};""", title="PE3R Demo") as demo:
     # scene state is save so that you can change conf_thr, cam_size... without rerunning the inference
-    scene = gradio.State(None)
+    # scene = gradio.State(None)
+
     gradio.HTML('<h2 style="text-align: center;">PE3R Demo</h2>')
     with gradio.Column():
         inputfiles = gradio.File(file_count="multiple")
@@ -587,7 +592,7 @@ with gradio.Blocks(css=""".gradio-container {margin: 0 !important; min-width: 10
 
         run_btn.click(fn=recon_fun,
                         inputs=[inputfiles],
-                        outputs=[scene, outmodel]) # , outgallery
+                        outputs=[outmodel]) # , outgallery, scene, 
         
         # find_btn.click(fn=get_3D_object_from_scene_fun,
         #                     inputs=[text_input, threshold, scene, min_conf_thr, as_pointcloud, mask_sky,