Test ZeroGPU

app.py

@@ -3,7 +3,21 @@ import gradio as gr
 import torch, spaces
 from diffusers import StableDiffusionPipeline, DPMSolverMultistepScheduler
 
-# ----- sanity: show CPU at startup (no GPU yet) -----
+# --- gradio_client bool-schema hotfix (safe no-op if not needed) ---
+try:
+    import gradio_client.utils as _gcu
+    _orig_get_type = _gcu.get_type
+    def _get_type_safe(schema):
+        # Handle JSON Schema booleans (e.g., additionalProperties: True/False)
+        if isinstance(schema, bool):
+            return "any"
+        return _orig_get_type(schema)
+    _gcu.get_type = _get_type_safe
+except Exception:
+    pass
+# -------------------------------------------------------------------
+
+# sanity: show CPU at startup (no GPU yet)
 zero = torch.tensor([0.0])
 print("startup device:", zero.device)  # should print: cpu
 
@@ -38,15 +52,14 @@ def snap8(x: int) -> int:
     x = max(256, min(1024, int(x)))
     return x - (x % 8)
 
-# ====== original ZeroGPU example ======
-@spaces.GPU
+@spaces.GPU  # tiny demo proving CUDA is present inside GPU block
 def greet(n: float):
     print("inside greet, cuda available:", torch.cuda.is_available())
     device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
     t = zero.to(device) + torch.tensor([float(n)], device=device)
     return f"Hello {t.item():.3f} Tensor (device: {t.device})"
 
-# ====== SD 1.5 text -> image ======
+# SD 1.5 text -> image
 @spaces.GPU(duration=120)
 def generate(prompt: str, negative: str, steps: int, cfg: float, width: int, height: int, seed: int):
     pipe = get_pipe()
@@ -72,7 +85,7 @@ def generate(prompt: str, negative: str, steps: int, cfg: float, width: int, hei
         )
     return out.images[0]
 
-# ====== UI (two tabs: sanity + SD) ======
+# ====== UI ======
 with gr.Blocks() as demo:
     gr.Markdown("# ZeroGPU demo + Stable Diffusion 1.5 (minimal)")
 
@@ -95,5 +108,10 @@ with gr.Blocks() as demo:
         )
 
 if __name__ == "__main__":
-    # keep it minimal so the Space "just builds"
-    demo.launch()
+    # On Spaces, expose a shareable URL & bind to 0.0.0.0
+    demo.queue(max_size=8).launch(
+        server_name="0.0.0.0",
+        server_port=7860,
+        share=True,
+        show_api=True,
+    )