Update mast3r/demo.py

mast3r/demo.py

@@ -15,7 +15,7 @@ import copy
 from scipy.spatial.transform import Rotation
 import tempfile
 import shutil
-+import spaces
+import spaces
 
 from mast3r.cloud_opt.sparse_ga import sparse_global_alignment
 from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
@@ -212,12 +212,13 @@ 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
-
+    
+@spaces.GPU
 def recon_fun(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
                                   silent, image_size):
-    functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
+    return functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
                                   silent, image_size)
-    +@spaces.GPU
+    
     
 def main_demo(tmpdirname, model, device, image_size, server_name, server_port, silent=False,
               share=False, gradio_delete_cache=False):