Count every model load, like from_pretrained

js/agate.js

@@ -63,14 +63,9 @@ export async function hasCachedModel() {
 export async function clearCache() { try { return await caches.delete(CACHE_NAME); } catch { return false; } }
 
 // The Hub counts a model download only on a request for the model repo's root config.json; the ONNX files
-// come from this Space, so each browser reads that config once (remembered in localStorage) to be counted.
-const COUNT_KEY = "agate-preview-001-counted";
-function countOnce() {
-  try { if (localStorage.getItem(COUNT_KEY)) return; } catch { /* storage blocked: still count this load */ }
-  fetch(COUNT_URL, { cache: "no-store" })
-    .then((r) => { if (r.ok) { try { localStorage.setItem(COUNT_KEY, String(Date.now())); } catch { /* ignore */ } } })
-    .catch(() => {});
-}
+// come from this Space, so every model load reads it once -- the same per-load count a Python
+// from_pretrained() produces, cached weights or not.
+function countLoad() { fetch(COUNT_URL, { cache: "no-store" }).catch(() => {}); }
 
 // ---- the model ---------------------------------------------------------------------------------
 export class Agate {
@@ -100,7 +95,7 @@ export class Agate {
       bufs[f] = buf;
     }
     this.stats = { downloadMB: total / 1e6, fromCacheMB: fromCache / 1e6, downloadMs: performance.now() - t0 };
-    countOnce();
+    countLoad();
     // drop cached files of older model versions (keys carry the sha256)
     try {
       const cache = await openCache(), keep = new Set(MODEL_FILES.map((f) => new URL(`${this.base}${real(f)}?sha256=${manifest.files[real(f)].sha256}`, location.href).href));