model-specific image for forward process

app.js

@@ -178,14 +178,16 @@ async function handleModelEvent(event) {
 function loadImage(image, label) { const canvas = document.createElement('canvas'); canvas.width = SIZE; canvas.height = SIZE; const context = canvas.getContext('2d'); const scale = Math.max(SIZE / image.width, SIZE / image.height); const width = image.width * scale; const height = image.height * scale; context.drawImage(image, (SIZE - width) / 2, (SIZE - height) / 2, width, height); sourcePixels = canvasPixels(canvas); imageLabel = label; reverseInputMode = 'seed'; reverseNoiseState = makeNoiseState(readSeed()); forwardRevealCount = 1; forwardCompleted = false; resetProgress('forward'); computeForwardStates(); renderTracks(); }
 modelInput.addEventListener('change', () => {
   applyTimesteps();
-  if (modelInput.value === 'google/ddpm-cifar10-32') {
-    const image = new Image();
-    image.onload = () => {
-      if (modelInput.value === 'google/ddpm-cifar10-32') loadImage(image, 'CIFAR-10 train 23');
-    };
-    image.onerror = () => setModelStatus('Could not load the CIFAR-10 source image.');
-    image.src = 'assets/cifar10_train_23_32.png';
-  }
+  const selectedModel = modelInput.value;
+  const source = selectedModel === 'google/ddpm-cifar10-32'
+    ? { path: 'assets/cifar10_train_23_32.png', label: 'CIFAR-10 train 23' }
+    : { path: 'assets/celeba_hq_validation_91_256.png', label: 'Human face' };
+  const image = new Image();
+  image.onload = () => {
+    if (modelInput.value === selectedModel) loadImage(image, source.label);
+  };
+  image.onerror = () => setModelStatus('Could not load the selected model’s source image.');
+  image.src = source.path;
 });
 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(); });

app.py

@@ -320,10 +320,11 @@ def change_source(filepath, seed, checkpoint_text, model_id=MODEL_ID):
 
 
 def change_model(seed, checkpoint_text, image, model_id):
-    if model_id == "google/ddpm-cifar10-32":
-        return change_source(ROOT / "assets/cifar10_train_23_32.png",
-                             seed, checkpoint_text, model_id)
-    return (image, *reset_tracks(seed, checkpoint_text, image, model_id))
+    source_paths = {
+        MODEL_ID: ROOT / "assets/celeba_hq_validation_91_256.png",
+        "google/ddpm-cifar10-32": ROOT / "assets/cifar10_train_23_32.png",
+    }
+    return change_source(source_paths[model_id], seed, checkpoint_text, model_id)
 
 
 def show_checkpoint_preview(gallery, event: gr.SelectData):