yansong1616 commited on
Commit
fbccfd3
1 Parent(s): 411a33c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -164,9 +164,9 @@ def get_args_parser():
164
  parser.add_argument("--server_port", type=int, help=("will start gradio app on this port (if available). "
165
  "If None, will search for an available port starting at 7860."),
166
  default=None)
167
- parser.add_argument("--weights", type=str, required=True, help="path to the model weights")
168
  parser.add_argument("--device", type=str, default='cpu', help="pytorch device")
169
- parser.add_argument("--tmp_dir", type=str, default=None, help="value for tempfile.tempdir")
170
  return parser
171
 
172
 
 
164
  parser.add_argument("--server_port", type=int, help=("will start gradio app on this port (if available). "
165
  "If None, will search for an available port starting at 7860."),
166
  default=None)
167
+ parser.add_argument("--weights", type=str, default="./checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", required=False, help="path to the model weights")
168
  parser.add_argument("--device", type=str, default='cpu', help="pytorch device")
169
+ parser.add_argument("--tmp_dir", type=str, default="./", help="value for tempfile.tempdir")
170
  return parser
171
 
172