FIx torch.save error. Add 3dgs tab.

app.py

@@ -9,7 +9,7 @@ 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
+from gs_demo import gs_demo_tab
 
 if __name__ == '__main__':
     # parser = get_args_parser()
@@ -32,8 +32,8 @@ if __name__ == '__main__':
             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)
+                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/mast3r_demo.py

@@ -6,6 +6,7 @@
 # sparse gradio demo functions
 # --------------------------------------------------------
 import sys
+import spaces
 
 import math
 import gradio
@@ -145,7 +146,7 @@ def get_3D_model_from_scene(silent, scene_state, min_conf_thr=2, as_pointcloud=F
     else:
         pts3d, _, confs = to_numpy(scene.get_dense_pts3d(clean_depth=clean_depth))
 
-    torch.save(confs, '/app/data/confs.pt')
+    # torch.save(confs, '/app/data/confs.pt')
     msk = to_numpy([c > min_conf_thr for c in confs])
     return _convert_scene_output_to_glb(outfile, rgbimg, pts3d, msk, focals, cams2world, as_pointcloud=as_pointcloud,
                                         transparent_cams=transparent_cams, cam_size=cam_size, silent=silent)
@@ -172,6 +173,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=300)
 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,