Improve Transformers.js code snippet

README.md

@@ -5,6 +5,7 @@ tags:
 - feature-extraction
 - sentence-similarity
 - mteb
+- transformers.js
 model-index:
 - name: binarize_False
   results:
@@ -3083,38 +3084,38 @@ Note the small differences compared to the full 768-dimensional similarities.
 
 ### Transformers.js
 
-```javascript
-import { pipeline } from '@xenova/transformers';
-
-// Create a feature extraction pipeline
-const extractor = await pipeline('feature-extraction', 'nomic-ai/modernbert-embed-base', {
-    quantized: false, // Comment out this line to use the quantized version
-});
-
-// Compute sentence embeddings
-const texts = ['search_query: What is TSNE?', 'search_query: Who is Laurens van der Maaten?'];
-const embeddings = await extractor(texts, { pooling: 'mean', normalize: true });
-console.log(embeddings);
+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
 ```
 
-<details><summary>Click to see Transformers.js usage with different quantizations</summary>
+Then, you can compute embeddings as follows:
 
 ```javascript
-import { pipeline } from '@xenova/transformers';
+import { pipeline, matmul } from '@huggingface/transformers';
 
 // Create a feature extraction pipeline
-const extractor = await pipeline('feature-extraction', 'nomic-ai/modernbert-embed-base', {
-    dtype: 'q4f16',
-});
-
-// Compute sentence embeddings
-const texts = ['search_query: What is TSNE?', 'search_query: Who is Laurens van der Maaten?'];
-const embeddings = await extractor(texts, { pooling: 'mean', normalize: true });
-console.log(embeddings);
-```
-
-</details>
+const extractor = await pipeline(
+  "feature-extraction",
+  "nomic-ai/modernbert-embed-base",
+  { dtype: "fp32" }, // Supported options: "fp32", "fp16", "q8", "q4", "q4f16"
+);
+
+// Embed queries and documents
+const query_embeddings = await extractor([
+    "search_query: What is TSNE?",
+    "search_query: Who is Laurens van der Maaten?",
+  ], { pooling: "mean", normalize: true },
+);
+const doc_embeddings = await extractor([
+    "search_document: TSNE is a dimensionality reduction algorithm created by Laurens van Der Maaten",
+  ], { pooling: "mean", normalize: true },
+);
 
+// Compute similarity scores
+const similarities = await matmul(query_embeddings, doc_embeddings.transpose(1, 0));
+console.log(similarities.tolist()); // [[0.721383273601532], [0.3259955644607544]]
+```
 
 
 ## Training
@@ -3152,4 +3153,4 @@ If you find the model, dataset, or training code useful, please cite our work
       archivePrefix={arXiv},
       primaryClass={cs.CL}
 }
-```
+```