zino36 commited on
Commit
8d8d6bc
·
verified ·
1 Parent(s): b226c36

Update mast3r/demo.py

Browse files
Files changed (1) hide show
  1. mast3r/demo.py +3 -3
mast3r/demo.py CHANGED
@@ -215,9 +215,9 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
215
 
216
  @spaces.GPU
217
  def recon_fun_decorator(tmpdirname, gradio_delete_cache, model, device,
218
- silent, image_size):
219
- return functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
220
- silent, image_size)
221
 
222
 
223
  def main_demo(tmpdirname, model, device, image_size, server_name, server_port, silent=False,
 
215
 
216
  @spaces.GPU
217
  def recon_fun_decorator(tmpdirname, gradio_delete_cache, model, device,
218
+ silent, image_size, *args):
219
+ return get_reconstructed_scene(tmpdirname, gradio_delete_cache, model, device,
220
+ silent, image_size, *args)
221
 
222
 
223
  def main_demo(tmpdirname, model, device, image_size, server_name, server_port, silent=False,