Detect ZeroGPU from Spaces environment

src/api/main.py

@@ -70,7 +70,14 @@ if _is_test_mode():
 
 
 def _is_zero_gpu_space() -> bool:
-    return os.environ.get("SPACE_ID", "").startswith("akagtag/")
+    hardware = os.environ.get("SPACE_HARDWARE", "").strip().lower()
+    zero_gpu_flag = os.environ.get("SPACES_ZERO_GPU", "").strip().lower()
+    space_id = os.environ.get("SPACE_ID", "").strip().lower()
+    return (
+        "zero-gpu" in hardware
+        or zero_gpu_flag in {"1", "true", "yes", "on"}
+        or space_id.startswith("akagtag/")
+    )
 
 
 def _effective_inference_backend():

tests/test_zero_gpu_contract.py

@@ -61,6 +61,8 @@ def test_api_gradio_compat_redirect_exists():
 
     assert "RedirectResponse" in source
     assert 'url="/"' in source
+    assert "SPACE_HARDWARE" in source
+    assert "SPACES_ZERO_GPU" in source
     assert any(
         isinstance(node, ast.AsyncFunctionDef) and node.name == "gradio_compat_redirect"
         for node in tree.body