init project

app.py

@@ -38,7 +38,7 @@ from modules.pe3r.models import Models
 import torchvision.transforms as tvf
 
 silent = False
-device = 'cpu' #'cuda' if torch.cuda.is_available() else 'cpu'
+device = 'cuda' if torch.cuda.is_available() else 'cpu' #'cpu' #
 pe3r = Models(device) #
 
 def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
@@ -503,6 +503,8 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
 
     outfile = get_3D_model_from_scene(outdir, scene, min_conf_thr, as_pointcloud, mask_sky,
                                       clean_depth, transparent_cams, cam_size)
+    
+    scene.to('cpu')
     torch.cuda.empty_cache()
 
     return scene, outfile