Add globals to avoid using inner declarations

app.py

@@ -2,12 +2,9 @@ import sys
 sys.path.append('wild-gaussian-splatting/mast3r/')
 sys.path.append('demo/')
 
-import os
-import tempfile
 import gradio as gr
 import torch
 from mast3r.demo import get_args_parser
-from mast3r.utils.misc import hash_md5
 from mast3r_demo import mast3r_demo_tab
 # from gs_demo import gs_demo_tab
 
@@ -18,25 +15,22 @@ if __name__ == '__main__':
     # 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'
+    # server_name = '0.0.0.0'# if args.local_network else '127.0.0.1'
 
     # weights_path = '/app/wild-gaussian-splatting/mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth'
-    weights_path = "naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"#args.weights if args.weights is not None else  + MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric
-    device = device = 'cuda' if torch.cuda.is_available() else 'cpu'
-    chkpt_tag = hash_md5(weights_path)
-
-    with tempfile.TemporaryDirectory(suffix='demo') as tmpdirname:
-        cache_path = os.path.join(tmpdirname, chkpt_tag)
-        os.makedirs(cache_path, exist_ok=True)
-
-        with gr.Blocks() as demo:
-            with gr.Tabs():
-                with gr.Tab("MASt3R Demo"):
-                    mast3r_demo_tab(cache_path, weights_path, device)
-                # with gr.Tab("Gaussian Splatting Demo"):
-                #     gs_demo_tab(cache_path)
-
-            demo.launch(show_error=True, share=None, server_name=None, server_port=None) 
-            # demo.launch(show_error=True, share=None, server_name='0.0.0.0', server_port=5555) 
+    # weights_path = "naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"#args.weights if args.weights is not None else  + MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric
+    # device = device = 'cuda' if torch.cuda.is_available() else 'cpu'
+    # chkpt_tag = hash_md5(weights_path)
+
+
+    with gr.Blocks() as demo:
+        with gr.Tabs():
+            with gr.Tab("MASt3R Demo"):
+                mast3r_demo_tab()
+            # with gr.Tab("Gaussian Splatting Demo"):
+            #     gs_demo_tab(cache_path)
+
+        demo.launch(show_error=True, share=None, server_name=None, server_port=None) 
+        # demo.launch(show_error=True, share=None, server_name='0.0.0.0', server_port=5555) 
 
 # python3 demo.py --weights "/app/mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth" --device "cuda" --server_port 3334 --local_network "$@"

demo/demo_globals.py

@@ -0,0 +1,19 @@
+import sys
+sys.path.append('wild-gaussian-splatting/mast3r/')
+
+import os
+import tempfile
+import torch
+from mast3r.utils.misc import hash_md5
+from mast3r.model import AsymmetricMASt3R
+
+weights_path = "naver/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric"
+weights_path = '/app/wild-gaussian-splatting/mast3r/checkpoints/MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric.pth'
+tmpdirname = tempfile.TemporaryDirectory(suffix='demo')
+chkpt_tag = hash_md5(weights_path)
+CACHE_PATH = os.path.join(tmpdirname.name, chkpt_tag)
+os.makedirs(CACHE_PATH, exist_ok=True)
+
+DEVICE = device = 'cuda' if torch.cuda.is_available() else 'cpu'
+MODEL = AsymmetricMASt3R.from_pretrained(weights_path).to(DEVICE)
+SILENT = False

demo/mast3r_demo.py

@@ -34,6 +34,8 @@ import matplotlib.pyplot as pl
 import torch
 
 
+from demo_globals import CACHE_PATH, MODEL, DEVICE, SILENT
+
 class SparseGAState():
     def __init__(self, sparse_ga, cache_dir=None, outfile_name=None):
         self.sparse_ga = sparse_ga
@@ -266,17 +268,16 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
     return win_col, winsize, win_cyclic, refid
 
 
-def mast3r_demo_tab(cache_path, weights_path, device, silent=False):
-    model = AsymmetricMASt3R.from_pretrained(weights_path).to(device)
-    
-    if not silent:
-        print('Outputing stuff in', cache_path)
+def get_reconstructed_scene_wrapper_func(*args, **kwargs):
+    return get_reconstructed_scene(CACHE_PATH, MODEL, DEVICE, SILENT, *args, **kwargs)
 
-    def get_reconstructed_scene_wrapper_func(*args, **kwargs):
-        return get_reconstructed_scene(cache_path, model, device, silent, *args, **kwargs)
+def update_3D_model_from_scene(*args, **kwargs):
+    return get_3D_model_from_scene(SILENT, *args, **kwargs)
 
-    def update_3D_model_from_scene(silent, *args, **kwargs):
-        return get_3D_model_from_scene(silent, *args, **kwargs)
+def mast3r_demo_tab():
+    
+    if not SILENT:
+        print('Outputing stuff in', CACHE_PATH)
 
     def get_context():
         css = """.gradio-container {margin: 0 !important; min-width: 100%};"""