Update mast3r/demo.py
Browse files- mast3r/demo.py +2 -2
mast3r/demo.py
CHANGED
@@ -214,7 +214,7 @@ def set_scenegraph_options(inputfiles, win_cyclic, refid, scenegraph_type):
|
|
214 |
return win_col, winsize, win_cyclic, refid
|
215 |
|
216 |
@spaces.GPU
|
217 |
-
def
|
218 |
silent, image_size):
|
219 |
return functools.partial(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
220 |
silent, image_size)
|
@@ -225,7 +225,7 @@ def main_demo(tmpdirname, model, device, image_size, server_name, server_port, s
|
|
225 |
if not silent:
|
226 |
print('Outputing stuff in', tmpdirname)
|
227 |
|
228 |
-
recon_fun =
|
229 |
silent, image_size)
|
230 |
model_from_scene_fun = functools.partial(get_3D_model_from_scene, silent)
|
231 |
|
|
|
214 |
return win_col, winsize, win_cyclic, refid
|
215 |
|
216 |
@spaces.GPU
|
217 |
+
def recon_fun_decorator(get_reconstructed_scene, 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)
|
|
|
225 |
if not silent:
|
226 |
print('Outputing stuff in', tmpdirname)
|
227 |
|
228 |
+
recon_fun = recon_fun_decorator(get_reconstructed_scene, tmpdirname, gradio_delete_cache, model, device,
|
229 |
silent, image_size)
|
230 |
model_from_scene_fun = functools.partial(get_3D_model_from_scene, silent)
|
231 |
|