kxhit commited on
Commit
21b0250
·
1 Parent(s): a71193c

landscape_only=False

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -527,7 +527,7 @@ def preview_input(inputfiles):
527
  silent = False
528
  image_size = 224
529
  weights_path = 'checkpoints/DUSt3R_ViTLarge_BaseDecoder_224_linear.pth'
530
- model = AsymmetricCroCo3DStereo.from_pretrained(weights_path).to(device)
531
  # dust3r will write the 3D model inside tmpdirname
532
  # with tempfile.TemporaryDirectory(suffix='dust3r_gradio_demo') as tmpdirname:
533
  tmpdirname = os.path.join('logs/user_object')
 
527
  silent = False
528
  image_size = 224
529
  weights_path = 'checkpoints/DUSt3R_ViTLarge_BaseDecoder_224_linear.pth'
530
+ model = AsymmetricCroCo3DStereo.from_pretrained(weights_path, landscape_only=False).to(device)
531
  # dust3r will write the 3D model inside tmpdirname
532
  # with tempfile.TemporaryDirectory(suffix='dust3r_gradio_demo') as tmpdirname:
533
  tmpdirname = os.path.join('logs/user_object')