arminak6 commited on
Commit
e362429
·
verified ·
1 Parent(s): 14beda5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +9 -2
app.py CHANGED
@@ -1,8 +1,15 @@
1
  from dust3r.demo import *
2
 
3
  if True:
4
- parser = get_args_parser()
5
- args = parser.parse_args()
 
 
 
 
 
 
 
6
 
7
  if args.tmp_dir is not None:
8
  tmp_path = args.tmp_dir
 
1
  from dust3r.demo import *
2
 
3
  if True:
4
+ # parser = get_args_parser()
5
+ # args = parser.parse_args()
6
+ args = {
7
+ "--weights": "./dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth",
8
+ "tmp_dir": None,
9
+ "server_name": None,
10
+ "device": "cuda",
11
+
12
+ }
13
 
14
  if args.tmp_dir is not None:
15
  tmp_path = args.tmp_dir