Another attempt to solve the pickling issues

demo.py

@@ -24,9 +24,11 @@ import main
 import utils.export as export
 
 @spaces.GPU(duration=15)
-def get_reconstructed_scene(outdir, model, silent, image_size, ios_mode, filelist):
+def get_reconstructed_scene(outdir, weights_path, silent, image_size, ios_mode, filelist):
 
+    # @TEMP: Temporarily instantiating the model here every time to avoid pickling issues with Hugging Face Spaces
     device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+    model = main.MAST3RGaussians.load_from_checkpoint(weights_path, device)
 
     assert len(filelist) == 1 or len(filelist) == 2, "Please provide one or two images"
     if ios_mode:
@@ -57,7 +59,6 @@ if __name__ == '__main__':
     model_name = "brandonsmart/splatt3r_v1.0"
     filename = "epoch=19-step=1200.ckpt"
     weights_path = hf_hub_download(repo_id=model_name, filename=filename)
-    model = main.MAST3RGaussians.load_from_checkpoint(weights_path, 'cpu')
     chkpt_tag = hash_md5(weights_path)
 
     # Define example inputs and their corresponding precalculated outputs
@@ -89,7 +90,7 @@ if __name__ == '__main__':
         cache_path = os.path.join(tmpdirname, chkpt_tag)
         os.makedirs(cache_path, exist_ok=True)
 
-        recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, model, silent, image_size, ios_mode)
+        recon_fun = functools.partial(get_reconstructed_scene, tmpdirname, weights_path, silent, image_size, ios_mode)
 
         if not ios_mode:
             for i in range(len(examples)):