zino36 commited on
Commit
fd1c52f
·
verified ·
1 Parent(s): 911025d

Update mast3r/demo.py

Browse files
Files changed (1) hide show
  1. mast3r/demo.py +1 -1
mast3r/demo.py CHANGED
@@ -15,6 +15,7 @@ import copy
15
  from scipy.spatial.transform import Rotation
16
  import tempfile
17
  import shutil
 
18
 
19
  from mast3r.cloud_opt.sparse_ga import sparse_global_alignment
20
  from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
@@ -214,7 +215,6 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
214
 
215
  def recon_fun(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
216
  silent, image_size):
217
- +import spaces
218
  functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
219
  silent, image_size)
220
  +@spaces.GPU
 
15
  from scipy.spatial.transform import Rotation
16
  import tempfile
17
  import shutil
18
+ +import spaces
19
 
20
  from mast3r.cloud_opt.sparse_ga import sparse_global_alignment
21
  from mast3r.cloud_opt.tsdf_optimizer import TSDFPostProcess
 
215
 
216
  def recon_fun(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
217
  silent, image_size):
 
218
  functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
219
  silent, image_size)
220
  +@spaces.GPU