Try inner functions

demo/mast3r_demo.py

@@ -174,6 +174,7 @@ def get_3D_model_from_scene(silent, scene_state, min_conf_thr=2, as_pointcloud=F
 #     save_pointcloud_with_normals(imgs, pts3d, masks, sparse_path)
 #     return save_path
 
+@spaces.GPU(duration=10)
 def get_reconstructed_scene(outdir, model, device, silent, image_size, current_scene_state,
                             filelist, optim_level, lr1, niter1, lr2, niter2, min_conf_thr, matching_conf_thr,
                             as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, scenegraph_type, winsize,
@@ -265,19 +266,18 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
     return win_col, winsize, win_cyclic, refid
 
 
-@spaces.GPU(duration=10)
-def get_reconstructed_scene_wrapper_func(cache_path, model, device, silent, *args, **kwargs):
-    return get_reconstructed_scene(cache_path, model, device, silent, *args, **kwargs)
-
-def get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs):
-    return get_3D_model_from_scene(silent, *args, **kwargs)
-
 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 update_3D_model_from_scene(silent, *args, **kwargs):
+        return get_3D_model_from_scene(silent, *args, **kwargs)
+
     def get_context():
         css = """.gradio-container {margin: 0 !important; min-width: 100%};"""
         title = "MASt3R Demo"
@@ -351,31 +351,31 @@ def mast3r_demo_tab(cache_path, weights_path, device, silent=False):
                 outputs=[scene, outmodel]
             )
             min_conf_thr.release(
-                fn=get_3D_model_from_scene_wrapper_func,
+                fn=update_3D_model_from_scene,
                 inputs=[scene, min_conf_thr, as_pointcloud, mask_sky, clean_depth, transparent_cams, cam_size, TSDF_thresh],
                 outputs=outmodel
             )
-            cam_size.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            cam_size.change(fn=update_3D_model_from_scene,
                             inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                     clean_depth, transparent_cams, cam_size, TSDF_thresh],
                             outputs=outmodel)
-            TSDF_thresh.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            TSDF_thresh.change(fn=update_3D_model_from_scene,
                                inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                        clean_depth, transparent_cams, cam_size, TSDF_thresh],
                                outputs=outmodel)
-            as_pointcloud.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            as_pointcloud.change(fn=update_3D_model_from_scene,
                                  inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                          clean_depth, transparent_cams, cam_size, TSDF_thresh],
                                  outputs=outmodel)
-            mask_sky.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            mask_sky.change(fn=update_3D_model_from_scene,
                             inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                     clean_depth, transparent_cams, cam_size, TSDF_thresh],
                             outputs=outmodel)
-            clean_depth.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            clean_depth.change(fn=update_3D_model_from_scene,
                                inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                        clean_depth, transparent_cams, cam_size, TSDF_thresh],
                                outputs=outmodel)
-            transparent_cams.change(fn=lambda *args, **kwargs: get_3D_model_from_scene_wrapper_func(silent, *args, **kwargs),
+            transparent_cams.change(fn=update_3D_model_from_scene,
                                     inputs=[scene, min_conf_thr, as_pointcloud, mask_sky,
                                             clean_depth, transparent_cams, cam_size, TSDF_thresh],
                                     outputs=outmodel)