- download_celeba_hq script

- change to docker project
- update README with Space metadata

Dockerfile

@@ -0,0 +1,15 @@
+FROM python:3.11-slim
+
+RUN useradd -m -u 1000 user
+WORKDIR /app
+
+COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt
+
+COPY --chown=user:user . .
+RUN mkdir -p /app/models && chown user:user /app/models
+USER user
+ENV HOME=/home/user
+
+EXPOSE 8000
+CMD ["python", "-u", "server.py"]

README.md

@@ -1,3 +1,13 @@
+---
+title: Diffusion
+emoji: 🌍
+colorFrom: indigo
+colorTo: purple
+sdk: docker
+app_port: 8000
+pinned: false
+---
+
 # Diffusion Lab
 
 The reverse process runs the pretrained `google/ddpm-celebahq-256` model through Hugging Face Diffusers. This is an unconditional CelebA-HQ face generator: reverse sampling starts with random Gaussian noise and does not denoise the uploaded image. The upload is used only by the browser-side forward-noising demonstration.
@@ -21,3 +31,10 @@ To generate and display one image without starting the server, run:
 python generate.py
 python generate.py --seed 1234
 ```
+
+To download a particular zero-based row from the public CelebA-HQ dataset:
+
+```sh
+python download_celeba_hq.py 1374
+python download_celeba_hq.py 42 --split train --output face-42.png
+```

app.js

@@ -161,4 +161,9 @@ function loadImage(image, label) { const canvas = document.createElement('canvas
 elements.applyTimesteps.addEventListener('click', applyTimesteps); elements.timestepsInput.addEventListener('keydown', (event) => { if (event.key === 'Enter') applyTimesteps(); });
 elements.randomizeSeed.addEventListener('click', () => { resampleNoise(); elements.configError.textContent = ''; forwardRevealCount = 1; forwardCompleted = false; resetProgress('forward'); computeForwardStates(); renderTracks(); });
 elements.imageUpload.addEventListener('change', (event) => { const file = event.target.files?.[0]; if (!file) return; const image = new Image(); image.onload = () => { loadImage(image, file.name.replace(/\.[^.]+$/, '')); URL.revokeObjectURL(image.src); }; image.src = URL.createObjectURL(file); });
+// Keep the original generated still life available while the default face loads,
+// and as a fallback if the image asset cannot be read.
 sourcePixels = canvasPixels(makeSourceArt()); computeForwardStates(); reverseNoiseState = makeNoiseState(readSeed()); resetProgress('forward'); resetProgress('reverse'); renderTracks();
+const defaultSourceImage = new Image();
+defaultSourceImage.onload = () => loadImage(defaultSourceImage, 'Human face');
+defaultSourceImage.src = 'assets/celeba_hq_validation_91_256.png';

download_celeba_hq.py

@@ -0,0 +1,80 @@
+import argparse
+import json
+from io import BytesIO
+from pathlib import Path
+from urllib.error import HTTPError, URLError
+from urllib.parse import urlencode
+from urllib.request import Request, urlopen
+
+from PIL import Image
+
+
+DATASET = "korexyz/celeba-hq-256x256"
+API_URL = "https://datasets-server.huggingface.co/rows"
+
+
+def download_row(row_index, split="validation", output=None):
+    """Download one CelebA-HQ row and save it as a 256x256 RGB PNG."""
+    if row_index < 0:
+        raise ValueError("Row index must be zero or greater.")
+
+    query = urlencode(
+        {
+            "dataset": DATASET,
+            "config": "default",
+            "split": split,
+            "offset": row_index,
+            "length": 1,
+        }
+    )
+    request = Request(f"{API_URL}?{query}", headers={"User-Agent": "diffusion-lab/1.0"})
+    with urlopen(request, timeout=30) as response:
+        result = json.load(response)
+
+    rows = result.get("rows", [])
+    if not rows:
+        total = result.get("num_rows_total", "the available")
+        raise ValueError(f"Row {row_index} is outside {split!r} ({total} rows).")
+
+    image_url = rows[0]["row"]["image"]["src"]
+    image_request = Request(image_url, headers={"User-Agent": "diffusion-lab/1.0"})
+    with urlopen(image_request, timeout=60) as response:
+        image_bytes = response.read()
+
+    destination = Path(output or f"celeba_hq_{split}_{row_index}_256.png")
+    #destination.parent.mkdir(parents=True, exist_ok=True)
+    with Image.open(BytesIO(image_bytes)) as image:
+        image = image.convert("RGB").resize((256, 256), Image.Resampling.LANCZOS)
+        image.save(destination, format="PNG", optimize=True)
+
+    return destination
+
+
+def main():
+    parser = argparse.ArgumentParser(
+        description="Download a specific row from the CelebA-HQ 256x256 dataset."
+    )
+    parser.add_argument("row_index", type=int, help="Zero-based row index")
+    parser.add_argument(
+        "--split",
+        choices=("train", "validation"),
+        default="validation",
+        help="Dataset split (default: validation)",
+    )
+    parser.add_argument(
+        "--output",
+        type=Path,
+        help="Output PNG path (default: celeba_hq_<split>_<row>_256.png)",
+    )
+    args = parser.parse_args()
+
+    try:
+        destination = download_row(args.row_index, args.split, args.output)
+    except (HTTPError, URLError, TimeoutError, ValueError, KeyError, OSError) as error:
+        parser.exit(1, f"Download failed: {error}\n")
+
+    print(f"Saved row {args.row_index} from {args.split!r} to {destination}")
+
+
+if __name__ == "__main__":
+    main()

index.html

@@ -4,7 +4,7 @@
   <meta charset="utf-8">
   <meta name="viewport" content="width=device-width, initial-scale=1">
   <meta name="theme-color" content="#f3f2ec">
-  <title>Diffusion Lab — A DDPM in motion</title>
+  <title>Diffusion — A DDPM in motion</title>
   <link rel="preconnect" href="https://fonts.googleapis.com">
   <link rel="preconnect" href="https://fonts.gstatic.com" crossorigin>
   <link href="https://fonts.googleapis.com/css2?family=DM+Mono:wght@400;500&family=DM+Sans:wght@400;500;600;700&family=Space+Grotesk:wght@400;500;600;700&display=swap" rel="stylesheet">
@@ -13,7 +13,8 @@
 <body>
   <div class="app-shell">
     <header class="topbar">
-      <a class="brand" href="#top"><span class="brand-mark" aria-hidden="true"><i></i><i></i><i></i><i></i></span><span>diffusion<span class="brand-light">lab</span></span></a>
+      <a class="brand" href="#top"><span class="brand-mark"
+					aria-hidden="true"><i></i><i></i><i></i><i></i></span><span>diffusion model</span></a>
       <div class="topbar-right"><span class="live-dot"></span></div>
     </header>
     <main id="top">
@@ -33,10 +34,6 @@
             <button id="randomize-seed" class="row-label-button" type="button" title="Generate a new random seed">Seed</button>
             <input id="seed-input" type="text" inputmode="numeric" value="42" spellcheck="false" aria-describedby="seed-help config-error">
           </div>
-          <div class="config-help">
-            <span id="timesteps-help">Timesteps: comma-separated values from 0 to 1000.</span>
-            <span id="seed-help">Select Seed for a new random value.</span>
-          </div>
           <p id="config-error" class="field-error" role="alert"></p>
         </div>
       </section>
@@ -44,7 +41,7 @@
         <div class="process-heading"><div class="panel-heading"><div>
 							<h2 id="diffusion-title">Forward process: image to noise</h2>
 					</div></div>
-					<span class="equation">x<sub>t</sub> = √α̅<sub>t</sub> x<sub>0</sub> + √(1 − α̅<sub>t</sub>) ε</span>
+					<span class="equation">x<sub>t</sub> = √α<sub>t</sub> x<sub>t−1</sub> + √(1 − α<sub>t</sub>) ε<sub>t</sub></span>
 				</div>
         <output id="forward-progress-label" class="step-progress">0/1000</output>
         <div id="diffusion-track" class="image-track" aria-label="Forward diffusion checkpoints"></div>

server.py

@@ -52,7 +52,12 @@ class DiffusionHandler(BaseHTTPRequestHandler):
             return
 
         relative_path = "index.html" if path == "/" else path.lstrip("/")
-        if relative_path not in {"index.html", "app.js", "styles.css"}:
+        if relative_path not in {
+            "index.html",
+            "app.js",
+            "styles.css",
+            "assets/celeba_hq_validation_91_256.png",
+        }:
             self.send_error(404)
             return
 
@@ -116,7 +121,7 @@ class DiffusionServer(ThreadingHTTPServer):
 
 
 if __name__ == "__main__":
-    server = DiffusionServer(("127.0.0.1", 8000), DiffusionHandler)
+    server = DiffusionServer(("0.0.0.0", 8000), DiffusionHandler)
     print(f"Diffusion Lab: http://{server.server_address[0]}:{server.server_address[1]}")
     print(f"Model: {MODEL_ID} (loaded on first generation request)")
     server.serve_forever()