Update mast3r/demo.py

mast3r/demo.py

@@ -212,7 +212,14 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
     refid = gradio.Slider(label="Scene Graph: Id", value=0, minimum=0,
                           maximum=num_files - 1, step=1, visible=scenegraph_type == 'oneref')
     return win_col, winsize, win_cyclic, refid
-    
+
+gtmpdirname = None
+ggradio_delete_cache = None
+gmodel = None
+gdevice = None
+gsilent = None
+gimage_size = None
+
 @spaces.GPU
 def recon_fun_decorator(tmpdirname, gradio_delete_cache, model, device,
                                   silent, image_size, *args, **kwargs):
@@ -224,7 +231,12 @@ def main_demo(tmpdirname, model, device, image_size, server_name, server_port, s
               share=False, gradio_delete_cache=False):
     if not silent:
         print('Outputing stuff in', tmpdirname)
-
+    gtmpdirname = tmpdirname
+    ggradio_delete_cache = gradio_delete_cache
+    gmodel = model
+    gdevice = device
+    gsilent = silent
+    gimage_size = image_size
     recon_fun = functools.partial(recon_fun_decorator, tmpdirname, gradio_delete_cache, model, device, silent, image_size)
     model_from_scene_fun = functools.partial(get_3D_model_from_scene, silent)