Spaces:
Running
on
Zero
Running
on
Zero
Remove argparse from demo
Browse files
app.py
CHANGED
@@ -5,21 +5,22 @@ sys.path.append('demo/')
|
|
5 |
import os
|
6 |
import tempfile
|
7 |
import gradio as gr
|
|
|
8 |
from mast3r.demo import get_args_parser
|
9 |
from mast3r.utils.misc import hash_md5
|
10 |
from mast3r_demo import mast3r_demo_tab
|
11 |
# from gs_demo import gs_demo_tab
|
12 |
|
13 |
if __name__ == '__main__':
|
14 |
-
parser = get_args_parser()
|
15 |
-
args = parser.parse_args()
|
16 |
|
17 |
-
if args.server_name is not None:
|
18 |
-
|
19 |
-
else:
|
20 |
-
|
21 |
|
22 |
-
weights_path = args.weights if args.weights is not None else
|
23 |
device = device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
24 |
chkpt_tag = hash_md5(weights_path)
|
25 |
|
|
|
5 |
import os
|
6 |
import tempfile
|
7 |
import gradio as gr
|
8 |
+
import torch
|
9 |
from mast3r.demo import get_args_parser
|
10 |
from mast3r.utils.misc import hash_md5
|
11 |
from mast3r_demo import mast3r_demo_tab
|
12 |
# from gs_demo import gs_demo_tab
|
13 |
|
14 |
if __name__ == '__main__':
|
15 |
+
# parser = get_args_parser()
|
16 |
+
# args = parser.parse_args()
|
17 |
|
18 |
+
# if args.server_name is not None:
|
19 |
+
# server_name = args.server_name
|
20 |
+
# else:
|
21 |
+
server_name = '0.0.0.0'# if args.local_network else '127.0.0.1'
|
22 |
|
23 |
+
weights_path = "naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"#args.weights if args.weights is not None else + MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric
|
24 |
device = device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
25 |
chkpt_tag = hash_md5(weights_path)
|
26 |
|