Update mast3r/demo.py

mast3r/demo.py

@@ -214,7 +214,7 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
     return win_col, winsize, win_cyclic, refid
     
 @spaces.GPU
-def recon_fun_decorator(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
+def recon_fun_decorator(tmpdirname, gradio_delete_cache, model, device,
                                   silent, image_size):
     return functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
                                   silent, image_size)
@@ -225,8 +225,7 @@ def main_demo(tmpdirname, model, device, image_size, server_name, server_port, s
     if not silent:
         print('Outputing stuff in', tmpdirname)
 
-    recon_fun = recon_fun_decorator(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
-                                  silent, 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)
 
     def get_context(delete_cache):