init project

app.py

@@ -40,6 +40,11 @@ import torchvision.transforms as tvf
 from modules.mast3r.model import AsymmetricMASt3R
 
 silent = False
+device = 'cpu'
+# pe3r = Models(device) 'cuda' if torch.cuda.is_available() else
+MAST3R_CKP = 'naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'
+mast3r = AsymmetricMASt3R.from_pretrained(MAST3R_CKP).to(device)
+
 
 def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
                                  cam_color=None, as_pointcloud=False,
@@ -442,11 +447,6 @@ def get_reconstructed_scene(outdir, filelist, schedule, niter, min_conf_thr,
     """
     if len(filelist) < 2:
         raise gradio.Error("Please input at least 2 images.")
-    
-    device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    # pe3r = Models(device)
-    MAST3R_CKP = 'naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'
-    mast3r = AsymmetricMASt3R.from_pretrained(MAST3R_CKP).to(device)
 
     images = Images(filelist=filelist, device=device)