Update index.html

index.html

@@ -42,14 +42,27 @@
     const hasWebGPU = 'gpu' in navigator;
     let device = hasWebGPU ? 'webgpu' : 'wasm';
     backendEl.textContent = device.toUpperCase();
-    envEl.textContent = hasWebGPU ? '✅ WebGPU detected (will fallback if slow)…'
-                                  : '⚠️ Using WASM (CPU).';
+    envEl.textContent = hasWebGPU ? '✅ WebGPU detected (will fallback if slow)…' : '⚠️ Using WASM (CPU).';
+
+    // ---------- SINGLE FIX: force ?download=1 on HF "resolve" URLs ----------
+    const _fetch = window.fetch.bind(window);
+    window.fetch = (url, opts) => {
+      try {
+        const u = new URL(url);
+        if (u.hostname === 'huggingface.co' && u.pathname.includes('/resolve/')) {
+          if (!u.searchParams.has('download')) u.searchParams.set('download', '1');
+          url = u.toString();
+        }
+      } catch {}
+      return _fetch(url, opts);
+    };
+    // -----------------------------------------------------------------------
 
     // Load Transformers.js v3
     const { pipeline } = await import('https://cdn.jsdelivr.net/npm/@huggingface/transformers@3');
 
     // Watchdog: if WebGPU load takes too long, retry on WASM
-    const LOAD_TIMEOUT_MS = 30000; // 30s
+    const LOAD_TIMEOUT_MS = 30000;
     let captioner;
 
     async function buildPipeline(targetDevice) {
@@ -90,7 +103,7 @@
       imgEl.src = imgURL;
     });
 
-    // Run captioning with better decoding (beam search)
+    // Run captioning (beam search for better quality)
     runBtn.addEventListener('click', async () => {
       if (!captioner) return;
       if (!imgURL) { logEl.textContent = 'Pick an image first.'; return; }
@@ -101,7 +114,7 @@
           num_beams: 5,
           do_sample: false,
           no_repeat_ngram_size: 3
-        }); // → [{ generated_text }]
+        });
         logEl.textContent = out[0].generated_text;
       } catch (e) {
         logEl.textContent = 'Inference error: ' + e;