Update index.js

index.js

@@ -1,54 +1,41 @@
-// Replace with the exact repo ID on HF Hub
-const MODEL_ID = 'onnx-community/orpheus-3b-0.1-ft-ONNX';
-
-let generator;
-
-async function init() {
-  // Show a loading message
-  const btn = document.getElementById('generate');
-  btn.disabled = true;
-  btn.textContent = 'Loading model…';
-
-  // Initialize the pipeline. This will download the tokenizer and ONNX files.
-  generator = await window.xenova.pipeline('text-generation', MODEL_ID);
-
-  btn.disabled = false;
-  btn.textContent = 'Generate';
-}
-
-async function generate() {
-  const promptEl = document.getElementById('prompt');
-  const outputEl = document.getElementById('output');
-  const btn = document.getElementById('generate');
-
-  const prompt = promptEl.value.trim();
-  if (!prompt) {
-    outputEl.textContent = 'Please enter a prompt.';
-    return;
-  }
-
-  btn.disabled = true;
-  btn.textContent = 'Generating…';
-  outputEl.textContent = '';
-
-  try {
-    // Call the ONNX‑powered transformer
-    const results = await generator(prompt, {
-      max_length: 200,
-      num_return_sequences: 1,
-      temperature: 0.7
-    });
-
-    // The library returns an array of sequences
-    outputEl.textContent = results[0].generated_text;
-  } catch (err) {
-    console.error(err);
-    outputEl.textContent = 'Error during generation. See console for details.';
-  } finally {
-    btn.disabled = false;
-    btn.textContent = 'Generate';
-  }
+import { pipeline, env } from "@xenova/transformers";
+
+async function main() {
+  const inputEl = document.getElementById("input-text");
+  const btnEl   = document.getElementById("generate-btn");
+  const outEl   = document.getElementById("output-text");
+
+  // (Optional) force ONNX CPU backend
+  env.backends.onnx.wasm.wasmPaths = "https://cdn.jsdelivr.net/npm/@xenova/transformers@2.17.2/dist/";
+  env.useWasmCPU();
+
+  // Load the text2text-generation pipeline with the ONNX weights
+  // hosted at onnx-community/orpheus-3b-0.1-ft-ONNX :contentReference[oaicite:0]{index=0}
+  const generator = await pipeline(
+    "text2text-generation",
+    "onnx-community/orpheus-3b-0.1-ft-ONNX"
+  );
+
+  // Enable button once ready
+  btnEl.disabled = false;
+
+  btnEl.addEventListener("click", async () => {
+    const prompt = inputEl.value.trim();
+    if (!prompt) return;
+
+    btnEl.disabled = true;
+    outEl.textContent = "Generating…";
+
+    try {
+      // Generate up to 256 tokens; adjust max_length as needed :contentReference[oaicite:1]{index=1}
+      const [res] = await generator(prompt, { max_length: 256 });
+      outEl.textContent = res.generated_text;
+    } catch (err) {
+      outEl.textContent = `Error: ${err}`;
+    } finally {
+      btnEl.disabled = false;
+    }
+  });
 }
 
-document.getElementById('generate').addEventListener('click', generate);
-window.addEventListener('load', init);
+main();