Update app.py

app.py

@@ -52,19 +52,19 @@ if __name__ == '__main__':
     parser = get_args_parser()
     args = parser.parse_args()
 
-    # Set default values for required arguments
-    if args.weights is None and args.model_name is None:
+    # Set default value for `args.weights` if not provided
+    if args.weights is None:
+        # Set a default model_name if weights are not provided
         args.model_name = 'MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'  # Default model_name
+        args.weights = "naver/" + args.model_name  # Construct default weights path
 
     if 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'
 
-    if args.weights is not None:
-        weights_path = args.weights
-    else:
-        weights_path = "naver/" + args.model_name
+    # Use the provided or default weights_path
+    weights_path = args.weights
 
     model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)
     chkpt_tag = hash_md5(weights_path)