Spaces:
Running
on
Zero
Running
on
Zero
Update app.py
Browse files
app.py
CHANGED
@@ -37,7 +37,7 @@ from dust3r.demo import _convert_scene_output_to_glb
|
|
37 |
def get_dust3r_args_parser():
|
38 |
parser = argparse.ArgumentParser()
|
39 |
parser.add_argument("--image_size", type=int, default=512, choices=[512, 224], help="image size")
|
40 |
-
parser.add_argument("--model_path", type=str, default="
|
41 |
parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
|
42 |
parser.add_argument("--batch_size", type=int, default=1)
|
43 |
parser.add_argument("--schedule", type=str, default='linear')
|
|
|
37 |
def get_dust3r_args_parser():
|
38 |
parser = argparse.ArgumentParser()
|
39 |
parser.add_argument("--image_size", type=int, default=512, choices=[512, 224], help="image size")
|
40 |
+
parser.add_argument("--model_path", type=str, default="naver/DUSt3R_ViTLarge_BaseDecoder_512_dpt", help="path to the model weights")
|
41 |
parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
|
42 |
parser.add_argument("--batch_size", type=int, default=1)
|
43 |
parser.add_argument("--schedule", type=str, default='linear')
|