deploy mast3r sfm

app.py

@@ -19,11 +19,15 @@ import mast3r.utils.path_to_dust3r  # noqa
 from dust3r.demo import set_print_with_timestamp
 
 import matplotlib.pyplot as pl
+
+import spaces
+
 pl.ion()
 
 torch.backends.cuda.matmul.allow_tf32 = True  # for gpu >= Ampere and pytorch >= 1.12
 
-if __name__ == '__main__':
+@spaces.GPU(duration=300)
+def main():
     parser = get_args_parser()
     args = parser.parse_args(['--model_name', 'MASt3R_ViTLarge_BaseDecoder_512_catmlpdpt_metric'])
     set_print_with_timestamp()
@@ -49,3 +53,6 @@ if __name__ == '__main__':
         os.makedirs(cache_path, exist_ok=True)
         main_demo(cache_path, model, args.retrieval_model, args.device, args.image_size, server_name, args.server_port,
                   silent=args.silent, share=args.share, gradio_delete_cache=args.gradio_delete_cache)
+
+if __name__ == '__main__':
+    main()

mast3r/demo.py

@@ -30,7 +30,6 @@ from dust3r.demo import get_args_parser as dust3r_get_args_parser
 
 import matplotlib.pyplot as pl
 
-import spaces
 
 class SparseGAState:
     def __init__(self, sparse_ga, should_delete=False, cache_dir=None, outfile_name=None):
@@ -253,7 +252,7 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
 
     return graph_opt, winsize, win_cyclic, refid
 
-@spaces.GPU(duration=300)
+
 def main_demo(tmpdirname, model, retrieval_model, device, image_size, server_name, server_port, silent=False,
               share=False, gradio_delete_cache=False):
     if not silent: