Deploy from GitHub Actions

README.md

@@ -1,13 +1,34 @@
 ---
 title: Cryptid Cannon
-emoji: 💻
-colorFrom: indigo
+emoji: 🔮
+colorFrom: purple
 colorTo: indigo
 sdk: gradio
-sdk_version: 6.18.0
-python_version: '3.13'
+sdk_version: 5.49.1
 app_file: app.py
 pinned: false
+license: apache-2.0
+suggested_hardware: zero-a10g
+short_description: Fire the cannon, summon a random absurd cryptid on klein 4B
+tags:
+  - build-small-hackathon
+  - thousand-token-wood
+  - tiny-titan
+  - flux
+  - text-to-image
 ---
 
-Check out the configuration reference at https://huggingface.co/docs/hub/spaces-config-reference
+# 🔮 Cryptid Cannon
+
+A deliberately silly [Build Small](https://huggingface.co/build-small-hackathon)
+toy. Press the button, summon a random absurd cryptid — a marshmallow golem mid-
+sneeze, a Victorian frog detective drinking tea — rendered by **FLUX.2 [klein]
+4B** (4 billion parameters, Apache 2.0, ~1s per image on the Space GPU).
+
+Leave the box empty for pure chaos, or whisper your own creature.
+
+This is a quick smoke-test of the deploy pipeline — not a serious competition
+entry. Built on the [klein starter](https://huggingface.co/spaces/stephenbtl/klein-build-small-starter).
+
+**Model:** [`FLUX.2-klein-4B`](https://huggingface.co/black-forest-labs/FLUX.2-klein-4B)
+· well under the hackathon's 32B cap (and under 4B-badge territory).

app.py

@@ -0,0 +1,127 @@
+"""
+🔮 Cryptid Cannon — fire the cannon, summon a random absurd cryptid.
+
+A deliberately silly FLUX.2 [klein] 4B toy for the Build Small hackathon.
+One button, one chaotic creature. Built on the klein starter's verified
+ZeroGPU + pipeline pattern.
+"""
+from __future__ import annotations
+
+import random
+import time
+
+# --- ZeroGPU shim: import `spaces` BEFORE torch -----------------------------
+try:
+    import spaces  # type: ignore
+
+    GPU = spaces.GPU
+except Exception:  # local / non-ZeroGPU fallback
+
+    def GPU(*dargs, **dkwargs):  # noqa: N802
+        if len(dargs) == 1 and callable(dargs[0]) and not dkwargs:
+            return dargs[0]
+
+        def wrap(fn):
+            return fn
+
+        return wrap
+
+
+import gradio as gr
+import torch
+from diffusers import Flux2KleinPipeline
+
+MODEL_ID = "black-forest-labs/FLUX.2-klein-4B"  # 4B, Apache 2.0, ungated
+STEPS = 4
+GUIDANCE = 1.0
+MAX_SEED = 2**31 - 1
+
+# Build on CPU at module scope (ZeroGPU pattern), but guard the load so the UI
+# still boots on tiny hardware instead of showing a blank runtime error.
+pipe = None
+LOAD_ERR = ""
+try:
+    print(f"Loading {MODEL_ID} on CPU…")
+    pipe = Flux2KleinPipeline.from_pretrained(MODEL_ID, torch_dtype=torch.bfloat16)
+    print("  loaded.")
+except Exception as e:  # noqa: BLE001
+    LOAD_ERR = str(e)
+    print("Model load failed:", e)
+
+CREATURES = [
+    "mushroom knight", "cloud cat", "disgruntled potato wizard", "sock goblin",
+    "philosophical jellyfish", "tiny bureaucrat dragon", "moss yeti",
+    "origami werewolf", "caffeinated snail", "Victorian frog detective",
+    "marshmallow golem", "interdimensional library owl",
+]
+TRAITS = [
+    "wearing a tiny hat", "made of stained glass", "on a unicycle",
+    "knitting a scarf", "looking very smug", "mid-sneeze", "holding a lantern",
+    "deeply confused", "in formal attire", "glowing faintly", "drinking tea",
+]
+STYLES = [
+    "children's storybook illustration, soft watercolor",
+    "claymation still, studio lighting",
+    "bold flat vector sticker art",
+    "cozy painted scene, warm light",
+    "1970s fantasy paperback cover",
+    "cute isometric 3D render",
+]
+
+
+def _roll(twist: str) -> str:
+    if twist and twist.strip():
+        subject = twist.strip()
+    else:
+        subject = f"{random.choice(CREATURES)} {random.choice(TRAITS)}"
+    return f"A {subject}, {random.choice(STYLES)}, centered, whimsical, no text"
+
+
+@GPU(duration=60)
+def summon(twist: str):
+    if pipe is None:
+        raise gr.Error(f"Model isn't loaded (this Space needs a GPU). {LOAD_ERR[:200]}")
+    prompt = _roll(twist)
+    pipe.to("cuda")
+    seed = random.randint(0, MAX_SEED)
+    t = time.time()
+    img = pipe(
+        prompt=prompt,
+        width=1024,
+        height=1024,
+        num_inference_steps=STEPS,
+        guidance_scale=GUIDANCE,
+        generator=torch.Generator(device="cuda").manual_seed(seed),
+    ).images[0]
+    return img, f"🔮 {prompt}\n· seed {seed} · {time.time() - t:.1f}s"
+
+
+with gr.Blocks(title="Cryptid Cannon", theme=gr.themes.Soft()) as demo:
+    gr.Markdown(
+        "# 🔮 Cryptid Cannon\n"
+        "Fire the cannon and summon a random absurd cryptid — powered by "
+        "**FLUX.2 [klein] 4B** (4B params, Apache 2.0, runs locally on the Space "
+        "GPU). Leave the box empty for pure chaos, or whisper your own creature."
+    )
+    with gr.Row():
+        with gr.Column():
+            twist = gr.Textbox(
+                label="Your creature (optional)",
+                placeholder="a grumpy mushroom knight",
+                lines=1,
+            )
+            btn = gr.Button("🔮 Fire the cannon", variant="primary")
+            gr.Examples(
+                ["", "a sleepy cloud cat", "a tiny bureaucrat dragon", "a philosophical jellyfish"],
+                twist,
+                label="Try one",
+            )
+        with gr.Column():
+            out = gr.Image(label="Your cryptid", height=480)
+            info = gr.Textbox(label="What you summoned", lines=3)
+    btn.click(summon, twist, [out, info])
+
+if __name__ == "__main__":
+    demo.queue(max_size=8).launch(
+        server_name="0.0.0.0", server_port=7860, show_error=True
+    )

requirements.txt

@@ -0,0 +1,15 @@
+# FLUX.2 [klein] needs a diffusers build that has Flux2KleinPipeline.
+# As of writing that's diffusers main, not a tagged release.
+git+https://github.com/huggingface/diffusers.git
+torch
+transformers
+accelerate
+# peft is the backend diffusers uses for load_lora_weights() — without it the
+# LoRA tab raises "PEFT backend is required for this method."
+peft
+safetensors
+sentencepiece
+protobuf
+pillow
+# Provides @spaces.GPU on ZeroGPU hardware. Harmless elsewhere.
+spaces