ostapagon commited on
Commit
52321ab
·
1 Parent(s): 7c39709

Remove argparse from demo

Browse files
Files changed (1) hide show
  1. app.py +8 -7
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
- server_name = args.server_name
19
- else:
20
- server_name = '0.0.0.0' if args.local_network else '127.0.0.1'
21
 
22
- weights_path = args.weights if args.weights is not None else "naver/" + args.model_name
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