Deploy OCR Model Workbench

app.py

@@ -27,6 +27,22 @@ ABSOLUTE_MAX_PAGES = int(os.getenv("ABSOLUTE_MAX_PAGES", "40"))
 STALE_RUN_HOURS = float(os.getenv("STALE_RUN_HOURS", "12"))
 
 
+try:
+    import spaces
+except Exception:
+    spaces = None
+
+
+def _gateway_probe() -> str:
+    return "ready"
+
+
+if spaces is not None:
+    gateway_zero_gpu_probe = spaces.GPU(duration=1)(_gateway_probe)
+else:
+    gateway_zero_gpu_probe = _gateway_probe
+
+
 def _spec(model_id: str) -> ModelSpec:
     try:
         return REGISTRY[model_id]

requirements.txt

@@ -1,5 +1,6 @@
 gradio>=5.49.1,<7.0
 gradio_client>=2.3.0,<3.0
+spaces>=0.40.0,<1.0
 httpx==0.28.1
 PyMuPDF==1.27.2.2
 Pillow==12.1.1