Update app.py

app.py

@@ -40,54 +40,29 @@
 # --------------------------------------------------------
 # gradio demo executable
 # --------------------------------------------------------
-import os
-import torch
-import tempfile
-from contextlib import nullcontext
-
-from mast3r.demo import get_args_parser, main_demo
-
-from mast3r.model import AsymmetricMASt3R
-from mast3r.utils.misc import hash_md5
-
-import matplotlib.pyplot as pl
-pl.ion()
-
-torch.backends.cuda.matmul.allow_tf32 = True  # for GPU >= Ampere and PyTorch >= 1.12
-
-def get_default_weights_path(model_name):
-    # Construct default weights path based on model_name
-    return f"naver/{model_name}"
-
 if __name__ == '__main__':
     parser = get_args_parser()
     args = parser.parse_args()
 
-    # Ensure at least one of weights or model_name is provided
+    # Set default values for required arguments
     if args.weights is None and args.model_name is None:
-        # Provide a default model_name if both weights and model_name are not provided
         args.model_name = 'MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'
     
-    # If weights are not provided but model_name is, construct weights_path
     if args.weights is None:
-        args.weights = get_default_weights_path(args.model_name)
-
-    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'
+        args.weights = f"naver/{args.model_name}"
 
-    # Use the provided or default weights_path
+    # Rest of the code for setting up the server and loading the model
+    server_name = args.server_name or ('0.0.0.0' if args.local_network else '127.0.0.1')
     weights_path = args.weights
 
-    # Load the model with the weights_path
+    # Load the model
     model = AsymmetricMASt3R.from_pretrained(weights_path).to(args.device)
     chkpt_tag = hash_md5(weights_path)
 
     def get_context(tmp_dir):
         return tempfile.TemporaryDirectory(suffix='_mast3r_gradio_demo') if tmp_dir is None \
             else nullcontext(tmp_dir)
-    
+
     with get_context(args.tmp_dir) as tmpdirname:
         cache_path = os.path.join(tmpdirname, chkpt_tag)
         os.makedirs(cache_path, exist_ok=True)