Switch dataset model to Qwen-Image + add 54-image training set (LFS)

scripts/generate_dataset.py:
- Use Qwen/Qwen-Image (fal-ai provider) — beat FLUX.1-dev on multi-animal
accuracy in side-by-side tests; FLUX.2-dev is edit-only so unusable for t2i
- Add a 4-attempt retry loop for transient fal-ai 504 timeouts
- Drop the --provider flag (single model now)

training/:
- 54 generated images + caption .txt files, tracked via Git LFS
- All 12 app animals solo, all 10 places, plus 2–3 animal and 2–3 place combos
- Captions share the app's exact build_subject prefix + NUMZOO_STYLE suffix

Co-Authored-By: Claude Opus 4.8 <noreply@anthropic.com>

.gitattributes

@@ -0,0 +1 @@
+training/*.jpg filter=lfs diff=lfs merge=lfs -text

scripts/generate_dataset.py

@@ -1,8 +1,9 @@
 """
 NumZoo training dataset generator.
 
-Generates LoRA training images matching the NumZoo aesthetic using FLUX.2-dev
+Generates LoRA training images matching the NumZoo aesthetic using Qwen-Image
 via the HuggingFace Inference API (fal-ai provider, billed to your HF Pro credits).
+(Qwen-Image beat FLUX.1-dev on multi-animal accuracy; FLUX.2-dev is edit-only.)
 
 Alignment with the live app is guaranteed by construction:
   - The "A cute {animals} {places}" prefix is built with image_generator.build_subject
@@ -18,10 +19,9 @@ Requirements:
     ~/miniforge3/bin/pip install huggingface_hub pillow python-dotenv
 
 Setup (HF Pro — just your existing token):
-    1. Accept the FLUX.2-dev license: https://huggingface.co/black-forest-labs/FLUX.2-dev
-    2. Get an HF token (fine-grained, "Make calls to Inference Providers" permission)
-       https://huggingface.co/settings/tokens
-    3. Add to .env:  HF_TOKEN=hf_...
+    1. Get an HF token (fine-grained, "Make calls to Inference Providers" permission)
+       https://huggingface.co/settings/tokens/new?ownUserPermissions=inference.serverless.write&tokenType=fineGrained
+    2. Add to .env:  HF_TOKEN=hf_...
 
 Usage:
     ~/miniforge3/bin/python3 scripts/generate_dataset.py            # all scenes
@@ -141,40 +141,43 @@ PROMPTS: list[tuple[str, str]] = [
 ]
 
 # ---------------------------------------------------------------------------
-# Generator using FLUX.2-dev via HuggingFace Inference API
+# Generator using Qwen-Image via HuggingFace Inference API (fal-ai provider)
 # ---------------------------------------------------------------------------
+# Qwen-Image chosen over FLUX.1-dev: far better at rendering DISTINCT animals in
+# multi-animal scenes (critical — the app lets players pick up to 3), and a
+# softer painterly storybook style closer to the NumZoo references.
+# Note: FLUX.2-dev is edit-only (image-to-image) on every HF provider, so it
+# cannot be used for text-to-image dataset generation.
 
-# Provider → model routing:
-#   fal-ai       → FLUX.2-dev      (best quality; billed to HF Pro credits)
-#   hf-inference → FLUX.1-schnell  (HF native fallback, lower quality)
-PROVIDERS = {
-    "fal-ai":       "black-forest-labs/FLUX.2-dev",
-    "hf-inference": "black-forest-labs/FLUX.1-schnell",
-}
+MODEL    = "Qwen/Qwen-Image"
+PROVIDER = "fal-ai"
+RETRIES  = 4   # fal-ai occasionally returns transient 504s
 
-def generate_image(prompt: str, provider: str = "fal-ai") -> "PIL.Image.Image":
+
+def generate_image(prompt: str) -> "PIL.Image.Image":
     from huggingface_hub import InferenceClient
 
-    # HF Pro token covers all providers — no separate provider key needed
+    # HF Pro token (with "Make calls to Inference Providers" permission) covers
+    # all providers — no separate provider key needed.
     hf_token = os.environ.get("HF_TOKEN")
-    model    = PROVIDERS.get(provider, PROVIDERS["fal-ai"])
-    client   = InferenceClient(provider=provider, api_key=hf_token)
-
-    image = client.text_to_image(
-        prompt,
-        model=model,
-        width=1024,
-        height=1024,
-    )
-    return image  # InferenceClient already returns a PIL Image
+    last_err = None
+    for attempt in range(1, RETRIES + 1):
+        try:
+            client = InferenceClient(provider=PROVIDER, api_key=hf_token)
+            return client.text_to_image(prompt, model=MODEL, width=1024, height=1024)
+        except Exception as e:
+            last_err = e
+            if attempt < RETRIES:
+                print(f"         ⚠️  attempt {attempt}/{RETRIES} failed ({str(e)[:60]}) — retrying")
+                time.sleep(4)
+    raise last_err  # exhausted retries
 
 
 def main():
     parser = argparse.ArgumentParser()
-    parser.add_argument("--start",    type=int,  default=1,         help="Resume from image N (1-based)")
-    parser.add_argument("--count",    type=int,  default=None,      help="Generate at most N images then stop")
-    parser.add_argument("--provider", type=str,  default="fal-ai",  help="Inference provider: fal-ai | hf-inference")
-    parser.add_argument("--dry-run",  action="store_true",          help="Print prompts without generating")
+    parser.add_argument("--start",    type=int,  default=1,    help="Resume from image N (1-based)")
+    parser.add_argument("--count",    type=int,  default=None, help="Generate at most N images then stop")
+    parser.add_argument("--dry-run",  action="store_true",     help="Print prompts without generating")
     args = parser.parse_args()
 
     if not args.dry_run:
@@ -217,7 +220,7 @@ def main():
             continue
 
         try:
-            image = generate_image(prompt, provider=args.provider)
+            image = generate_image(prompt)
             image.save(img_path, "JPEG", quality=95)
             txt_path.write_text(prompt)
             generated_this_run += 1

training/image_001.txt

@@ -0,0 +1 @@
+A cute bunny in an enchanted mushroom forest, sitting on a polka-dot toadstool, fireflies and floating spores drifting around, soft lantern glow, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_002.txt

@@ -0,0 +1 @@
+A cute kitten on a sunny beach with ocean waves, building a tiny sandcastle with a bucket and shells, gentle waves lapping, warm sunset sky, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_003.txt

@@ -0,0 +1 @@
+A cute puppy in a cosy cottage garden, napping in a flower-filled wheelbarrow, watering can and butterflies nearby, golden afternoon light, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_004.txt

@@ -0,0 +1 @@
+A cute baby fox surrounded by sparkling stars, curled up on a fluffy cloud cradling a tiny glowing star, glittering night sky, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_005.txt

@@ -0,0 +1 @@
+A cute baby panda in a cherry blossom garden, nibbling a dango skewer as petals fall, paper lanterns strung above, soft pink light, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_006.txt

@@ -0,0 +1 @@
+A cute baby koala on a tropical island, hugging a palm trunk with a coconut drink, striped hammock and a parrot, turquoise sea behind, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_007.txt

@@ -0,0 +1 @@
+A cute baby lion under a rainbow, wearing a tiny crown at the end of a rainbow, pastel clouds and floating sparkles, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_008.txt

@@ -0,0 +1 @@
+A cute baby tiger on a snowy mountain top, bundled in a knitted scarf on a snowy peak planting a tiny flag, sparkling snow and faint aurora, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_009.txt

@@ -0,0 +1 @@
+A cute baby frog in a field of tropical flowers, perched on a giant hibiscus bloom, dewdrops glistening, big tropical leaves and warm bokeh, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_010.txt

@@ -0,0 +1 @@
+A cute baby penguin on a glowing crescent moon, sitting on the curve of a glowing crescent moon in a knitted hat, scattered twinkling stars, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_011.txt

@@ -0,0 +1 @@
+A cute butterfly in a cherry blossom garden, fluttering through cherry blossoms trailing sparkles, pastel petals swirling in the breeze, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_012.txt

@@ -0,0 +1 @@
+A cute unicorn surrounded by sparkling stars, galloping across a starry sky, rainbow mane glowing, a trail of sparkles behind, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_013.txt

@@ -0,0 +1 @@
+A cute puppy in an enchanted mushroom forest, exploring beneath a giant mushroom with a tiny lantern, glowing toadstools and soft moss, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_014.txt

@@ -0,0 +1 @@
+A cute kitten on a glowing crescent moon, curled asleep on a crescent moon wearing a nightcap, twinkling stars all around, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_015.txt

@@ -0,0 +1 @@
+A cute bunny on a sunny beach with ocean waves, splashing in shallow waves beside a starfish friend, beach pail and spade, pink sunset, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_016.txt

@@ -0,0 +1 @@
+A cute baby panda in a cosy cottage garden, tending a vegetable patch in a straw hat, bees and tall sunflowers, warm sun, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_017.txt

@@ -0,0 +1 @@
+A cute baby penguin on a snowy mountain top, sliding down a snowy slope on its belly, scarf flying, sparkling powder snow, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_018.txt

@@ -0,0 +1 @@
+A cute baby fox on a tropical island, lounging in a hammock between two palms with sunglasses, coconuts and calm ocean, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_019.txt

@@ -0,0 +1 @@
+A cute baby lion in a field of tropical flowers, snoozing in a field of tropical flowers, a butterfly on its nose, dappled golden light, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_020.txt

@@ -0,0 +1 @@
+A cute baby tiger in a cherry blossom garden, chasing falling cherry petals, paper lanterns above, soft pink and lilac tones, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_021.txt

@@ -0,0 +1 @@
+A cute baby frog in an enchanted mushroom forest, playing a tiny flute on a lily pad among glowing mushrooms, fireflies and reeds, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_022.txt

@@ -0,0 +1 @@
+A cute unicorn under a rainbow, standing proudly under a rainbow, flower garland around its neck, pastel clouds, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_023.txt

@@ -0,0 +1 @@
+A cute bunny and kitten in an enchanted mushroom forest, roasting marshmallows over a tiny campfire, fireflies and glowing mushrooms, cozy night, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text

training/image_024.txt

@@ -0,0 +1 @@
+A cute puppy and baby fox on a sunny beach with ocean waves, building a sandcastle together with shell flags, gentle waves at golden hour, kawaii children's book illustration, pastel anime art style, soft painterly lighting, detailed rich background with warm fairy lights, cozy magical atmosphere, cute chibi character with big sparkling eyes, soft pastel color palette, highly detailed scene, no text