Update app.py

app.py

@@ -3,26 +3,26 @@ from dust3r.demo import *
 if True:
     # parser = get_args_parser()
     # args = parser.parse_args()
-    args = {
-        "--weights": "./dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth",
-        "tmp_dir": None,
-        "server_name": None,
-        "device": "cuda",
+    # args = {
+    #     "--weights": "./dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth",
+    #     "tmp_dir": None,
+    #     "server_name": None,
+    #     "device": "cuda",
         
-    }
+    # }
 
-    if args.tmp_dir is not None:
-        tmp_path = args.tmp_dir
-        os.makedirs(tmp_path, exist_ok=True)
-        tempfile.tempdir = tmp_path
+    # if args.tmp_dir is not None:
+    #     tmp_path = args.tmp_dir
+    #     os.makedirs(tmp_path, exist_ok=True)
+    #     tempfile.tempdir = tmp_path
 
-    if args.server_name is not None:
+    if False: #args.server_name is not None:
         server_name = args.server_name
     else:
-        server_name = '0.0.0.0' if args.local_network else '127.0.0.1'
+        server_name = '0.0.0.0' # if args.local_network else '127.0.0.1'
 
-    model = load_model(args.weights, args.device)
+    model = load_model("./dust3r/checkpoints/DUSt3R_ViTLarge_BaseDecoder_512_dpt.pth", "cuda")
     # dust3r will write the 3D model inside tmpdirname
     with tempfile.TemporaryDirectory(suffix='dust3r_gradio_demo') as tmpdirname:
         print('Outputing stuff in', tmpdirname)
-        main_demo(tmpdirname, model, args.devie, args.image_size, server_name, args.server_port)
+        main_demo(tmpdirname, model, "cuda", 512, server_name, 7860)