init project

app.py

@@ -37,6 +37,10 @@ import matplotlib.pyplot as pl
 from modules.mobilesamv2.utils.transforms import ResizeLongestSide
 from modules.pe3r.models import Models
 
+silent = True
+device = 'cuda' if torch.cuda.is_available() else 'cpu'
+pe3r = Models(device)
+
 def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
                                  cam_color=None, as_pointcloud=False,
                                  transparent_cams=False, silent=False):
@@ -552,13 +556,6 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
     return winsize, refid
 
 
-
-
-
-silent = True
-device = 'cuda' if torch.cuda.is_available() else 'cpu'
-pe3r = Models(device)
-
 with tempfile.TemporaryDirectory(suffix='pe3r_gradio_demo') as tmpdirname:
     recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)
     model_from_scene_fun = functools.partial(get_3D_model_from_scene, tmpdirname, silent)

modules/pe3r/demo.py

@@ -550,7 +550,6 @@ def set_scenegraph_options(inputfiles, winsize, refid, scenegraph_type):
 
 def main_demo(tmpdirname, server_name, server_port, silent=False):
     device = 'cuda' if torch.cuda.is_available() else 'cpu'
-
     pe3r = Models(device)
     
     recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, pe3r, device, silent)