yansong1616 commited on
Commit
eeb12d4
·
verified ·
1 Parent(s): 9f2fd76

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -68,7 +68,7 @@ def get_args_parser():
68
  parser.add_argument("--server_port", type=int, help=("will start gradio app on this port (if available). "
69
  "If None, will search for an available port starting at 7860."),
70
  default=None)
71
- parser.add_argument("--weights", type=str, required=True, help="path to the model weights")
72
  parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
73
  parser.add_argument("--tmp_dir", type=str, default=None, help="value for tempfile.tempdir")
74
  return parser
 
68
  parser.add_argument("--server_port", type=int, help=("will start gradio app on this port (if available). "
69
  "If None, will search for an available port starting at 7860."),
70
  default=None)
71
+ parser.add_argument("--weights", type=str, default="./checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", required=False, help="path to the model weights")
72
  parser.add_argument("--device", type=str, default='cuda', help="pytorch device")
73
  parser.add_argument("--tmp_dir", type=str, default=None, help="value for tempfile.tempdir")
74
  return parser