Fuzail22's picture
Update README.md
93f4405 verified
|
raw
history blame
942 Bytes
---
license: mit
base_model:
- sentence-transformers/msmarco-distilbert-cos-v5
---
# Usage using Transformer.js for feature extraction
```
import { env, pipeline } from "@xenova/transformers";
env.allowLocalModels = false;
const extractor = await pipeline(
"feature-extraction",
"Fuzail22/onnx-msmarco-distilbert-cos-v5",
{
quantized: false
}
);
const extract = async (text: string) => {
const output = await extractor(text, {
pooling: "mean",
normalize: true
});
console.log("extraction output:", output);
};
extact("Open Source is great!")
```
> This is Onnx converted version of the original model
> The Input Sequence length is 512
Converted Using [Xenova](https://github.com/xenova/transformers.js/blob/main/scripts/convert.py)
Link to original Model [sentence-transformers/msmarco-distilbert-cos-v5](https://huggingface.co/sentence-transformers/msmarco-distilbert-cos-v5/blob/main/README.md)