junaid-simamdigital commited on
Commit
8b048ca
·
verified ·
1 Parent(s): 8a9129c

Decouple deterministic smoke from GPU quota

Browse files
Files changed (1) hide show
  1. app.py +13 -2
app.py CHANGED
@@ -633,8 +633,7 @@ def recent_failure_reports():
633
  return [str(path) for path in sorted(OUTPUT_DIR.glob("*_failure.json"), key=lambda item: item.stat().st_mtime, reverse=True)[:10]]
634
 
635
 
636
- @GPU_FUNCTION
637
- def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
638
  if image is None:
639
  raise gr.Error("Please add an input image first.")
640
  started = time.perf_counter()
@@ -840,6 +839,18 @@ def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mod
840
  ) from exc
841
 
842
 
 
 
 
 
 
 
 
 
 
 
 
 
843
  with gr.Blocks(title="Simam3D") as demo:
844
  gr.Markdown("# Simam3D\nSingle-image depth and 3D reconstruction workbench")
845
  gr.Markdown(
 
633
  return [str(path) for path in sorted(OUTPUT_DIR.glob("*_failure.json"), key=lambda item: item.stat().st_mtime, reverse=True)[:10]]
634
 
635
 
636
+ def _generate_impl(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
 
637
  if image is None:
638
  raise gr.Error("Please add an input image first.")
639
  started = time.perf_counter()
 
839
  ) from exc
840
 
841
 
842
+ @GPU_FUNCTION
843
+ def _generate_gpu(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
844
+ return _generate_impl(image, extra_files, backend, guess, demo_mode, progress)
845
+
846
+
847
+ def generate(image: Image.Image, extra_files, backend: str, guess: str, demo_mode: bool, progress=gr.Progress()):
848
+ """Keep the deterministic export diagnostic independent of GPU quota."""
849
+ if demo_mode:
850
+ return _generate_impl(image, extra_files, backend, guess, demo_mode, progress)
851
+ return _generate_gpu(image, extra_files, backend, guess, demo_mode, progress)
852
+
853
+
854
  with gr.Blocks(title="Simam3D") as demo:
855
  gr.Markdown("# Simam3D\nSingle-image depth and 3D reconstruction workbench")
856
  gr.Markdown(