Upload README.md

README.md

@@ -112,6 +112,73 @@ for i, query in enumerate(queries):
 #  Similarity: 0.4238 | Document 0: Machine learning is a subset of ...
 #  Similarity: 0.5723 | Document 1: Neural networks are trained ...
 ```
+
+## Transformers.js Usage
+
+If you haven't already, you can install the [Transformers.js](https://huggingface.co/docs/transformers.js) JavaScript library from [NPM](https://www.npmjs.com/package/@huggingface/transformers) using:
+```bash
+npm i @huggingface/transformers
+```
+
+You can then use the model to compute embeddings like this:
+
+```js
+import { AutoModel, AutoTokenizer, matmul } from "@huggingface/transformers";
+
+// Download from the 🤗 Hub
+const model_id = "onnx-community/mdbr-leaf-ir-ONNX";
+const tokenizer = await AutoTokenizer.from_pretrained(model_id);
+const model = await AutoModel.from_pretrained(model_id, {
+    dtype: "fp32", // Options: "fp32" | "q8" | "q4"
+});
+
+// Prepare queries and documents
+const queries = [
+    "What is machine learning?",
+    "How does neural network training work?",
+];
+const documents = [  
+    "Machine learning is a subset of artificial intelligence that focuses on algorithms that can learn from data.",
+    "Neural networks are trained through backpropagation, adjusting weights to minimize prediction errors.",
+];
+const inputs = await tokenizer([
+    ...queries.map((x) => "Represent this sentence for searching relevant passages: " + x),
+    ...documents,
+], { padding: true });
+
+// Generate embeddings
+const { sentence_embedding } = await model(inputs);
+
+// Compute similarities
+const scores = await matmul(
+    sentence_embedding.slice([0, queries.length]),
+    sentence_embedding.slice([queries.length, null]).transpose(1, 0),
+);
+const scores_list = scores.tolist();
+
+for (let i = 0; i < queries.length; ++i) {
+    console.log(`Query: ${queries[i]}`);
+    for (let j = 0; j < documents.length; ++j) {
+        console.log(` Similarity: ${scores_list[i][j].toFixed(4)} | Document ${j}: ${documents[j]}`);
+    }
+    console.log();
+}
+```
+
+<details>
+
+<summary>See example output</summary>
+
+```
+Query: What is machine learning?
+ Similarity: 0.6857 | Document 0: Machine learning is a subset of artificial intelligence that focuses on algorithms that can learn from data.
+ Similarity: 0.4598 | Document 1: Neural networks are trained through backpropagation, adjusting weights to minimize prediction errors.
+
+Query: How does neural network training work?
+ Similarity: 0.4238 | Document 0: Machine learning is a subset of artificial intelligence that focuses on algorithms that can learn from data.
+ Similarity: 0.5723 | Document 1: Neural networks are trained through backpropagation, adjusting weights to minimize prediction errors.
+```
+</details>
   
 ## Transformers Usage