Deploy GeoAP YOLO26m detection + segmentation Space

app.py

@@ -29,6 +29,21 @@ from src.viz import hex_for
 logging.basicConfig(level=logging.INFO, format="[%(asctime)s] %(levelname)s %(name)s — %(message)s")
 logger = logging.getLogger("geoap.space")
 
+# ── ZeroGPU ──────────────────────────────────────────────────────────────────
+# On ZeroGPU hardware a GPU is attached only for the duration of a function
+# marked with @spaces.GPU. The `spaces` package exists only on HF, so locally
+# we fall back to a no-op decorator and nothing changes.
+try:
+    import spaces  # type: ignore
+
+    def gpu(duration: int = 120):
+        return spaces.GPU(duration=duration)
+except ImportError:  # local run / non-ZeroGPU Space
+    def gpu(duration: int = 120):
+        def _identity(fn):
+            return fn
+        return _identity
+
 # Gradio 6 moved `theme` from Blocks(...) to launch(...). Support both.
 GRADIO_MAJOR = int(gr.__version__.split(".")[0])
 _THEME_ON_LAUNCH = GRADIO_MAJOR >= 6
@@ -101,6 +116,7 @@ def _no_input():
     return None, [], {"error": "Upload an image first."}, None, PLACEHOLDER
 
 
+@gpu(duration=120)
 def predict_detection(raster, conf, iou, tile_size, overlap, progress=gr.Progress()):
     if raster is None:
         return _no_input()
@@ -112,6 +128,7 @@ def predict_detection(raster, conf, iou, tile_size, overlap, progress=gr.Progres
     return _finish(res, raster, S.detection_classes, DET_COLORS, "detection")
 
 
+@gpu(duration=180)
 def predict_segmentation(raster, conf, iou, tile_size, overlap, progress=gr.Progress()):
     if raster is None:
         return _no_input()

requirements.txt

@@ -11,9 +11,10 @@ pyyaml
 jinja2>=3.1.2               # pandas/Styler backend used by gr.Dataframe
 rasterio>=1.3               # GeoTIFF reading + CRS reprojection for the map
 shapely>=2.0
-# CPU-only torch: PyPI's default linux wheel bundles CUDA (~2.5GB) which is
-# dead weight on cpu-basic hardware. The +cpu local version sorts higher than
-# the plain PyPI release, so pip prefers it from this index.
---extra-index-url https://download.pytorch.org/whl/cpu
+# ZeroGPU: allocates a GPU only inside @spaces.GPU-decorated functions.
+spaces>=0.30.0
+# Keep the default (CUDA) torch wheels — ZeroGPU needs them. If you ever move
+# this Space to cpu-basic/cpu-upgrade, add:
+#   --extra-index-url https://download.pytorch.org/whl/cpu
 torch
 torchvision

scripts/deploy_space.py

@@ -27,6 +27,8 @@ from src.settings import ROOT, load_settings  # noqa: E402
 BASE_IGNORE: List[str] = [
     ".venv/*", "venv/*", "__pycache__/*", "*.pyc", ".git/*", ".gitignore",
     "runs/*", ".DS_Store", "*.log", ".ipynb_checkpoints/*", "flagged/*",
+    # secrets — upload_folder does NOT read .gitignore, so list them here too
+    "token.txt", "*.token", ".env", ".env.*", "secrets*", "*credentials*",
 ]
 WEIGHT_IGNORE: List[str] = ["weights/*.pt", "weights/*.onnx", "weights/*.engine"]