lixi042 commited on
Commit
7ebe1be
·
1 Parent(s): 25eddc4

Add @spaces.GPU decorator for ZeroGPU compatibility

Browse files
Files changed (1) hide show
  1. app.py +2 -0
app.py CHANGED
@@ -365,6 +365,7 @@ def update_gallery_on_upload(input_images):
365
 
366
 
367
  # -------------------------- 3D Reconstruction Pipeline --------------------------
 
368
  def gradio_demo(
369
  target_dir,
370
  conf_thres=5.0,
@@ -726,6 +727,7 @@ def get_scene_info(examples_dir):
726
  return scenes
727
 
728
 
 
729
  def example_pipeline(
730
  scene,
731
  conf_thres=5.0,
 
365
 
366
 
367
  # -------------------------- 3D Reconstruction Pipeline --------------------------
368
+ @spaces.GPU
369
  def gradio_demo(
370
  target_dir,
371
  conf_thres=5.0,
 
727
  return scenes
728
 
729
 
730
+ @spaces.GPU
731
  def example_pipeline(
732
  scene,
733
  conf_thres=5.0,