Upload example_web.js with huggingface_hub

example_web.js

@@ -12,7 +12,7 @@
 */
 
 const REPO = 'Bartholomheow/Supra2-IMG-ONNX';
-const fileUrl = (path) => `https://huggingface.co/${REPO}/resolve/main/${path}`;
+const fileUrl = (repo, path) => `https://huggingface.co/${repo}/resolve/main/${path}`;
 
 function mulberry32(seed) {
   let a = seed >>> 0;
@@ -78,14 +78,14 @@ function halfToFloat(h) {
   return f[0];
 }
 
-export async function loadSupra({ onProgress } = {}) {
+export async function loadSupra(repo = REPO, { onProgress } = {}) {
   const ort = await import('onnxruntime-web');
   const { AutoTokenizer } = await import('@huggingface/transformers');
-  const cfg = await (await fetch(fileUrl('pipeline_config.json'))).json();
+  const cfg = await (await fetch(fileUrl(repo, 'pipeline_config.json'))).json();
   const [ditBuf, encBuf, vaeBuf] = await Promise.all([
-    download(fileUrl(cfg.dit), cfg.dit, onProgress),
-    download(fileUrl(cfg.text_encoder), cfg.text_encoder, onProgress),
-    download(fileUrl(cfg.vae_decoder), cfg.vae_decoder, onProgress),
+    download(fileUrl(repo, cfg.dit), cfg.dit, onProgress),
+    download(fileUrl(repo, cfg.text_encoder), cfg.text_encoder, onProgress),
+    download(fileUrl(repo, cfg.vae_decoder), cfg.vae_decoder, onProgress),
   ]);
   const opts = { executionProviders: ['webgpu'] };
   const [dit, enc, vae] = await Promise.all([
@@ -93,7 +93,7 @@ export async function loadSupra({ onProgress } = {}) {
     ort.InferenceSession.create(encBuf, opts),
     ort.InferenceSession.create(vaeBuf, opts),
   ]);
-  const tokenize = await AutoTokenizer.from_pretrained(REPO);
+  const tokenize = await AutoTokenizer.from_pretrained(repo);
   return { ort, cfg, dit, enc, vae, tokenize };
 }
 
@@ -165,6 +165,6 @@ export function paint(pixels, size, canvas) {
 }
 
 // Usage:
-// const model = await loadSupra({ onProgress: (p) => console.log(p.file, p.loaded) });
+// const model = await loadSupra(REPO, { onProgress: (p) => console.log(p.file, p.loaded) });
 // const { pixels, size } = await generate(model, 'a lighthouse above violet clouds at dusk');
 // paint(pixels, size, document.querySelector('canvas'));