deploy mast3r sfm

mast3r/demo.py

@@ -30,6 +30,7 @@ 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):
@@ -252,7 +253,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: