init project

app.py

@@ -40,12 +40,6 @@ import torchvision.transforms as tvf
 from modules.mast3r.model import AsymmetricMASt3R
 
 silent = False
-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)
-print(device)
-
 
 def _convert_scene_output_to_glb(outdir, imgs, pts3d, mask, focals, cams2world, cam_size=0.05,
                                  cam_color=None, as_pointcloud=False,
@@ -448,6 +442,11 @@ 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)