Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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=
|
| 168 |
parser.add_argument("--device", type=str, default='cpu', help="pytorch device")
|
| 169 |
-
parser.add_argument("--tmp_dir", type=str, default=
|
| 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 |
|