https://huggingface.co/tasksource/deberta-small-long-nli with ONNX weights to be compatible with Transformers.js.
Usage (Transformers.js)
If you haven't already, you can install the Transformers.js JavaScript library from NPM using:
npm i @huggingface/transformers
You can then use the model for zero-shot classification as follows:
import { pipeline } from '@huggingface/transformers';
// Create a zero-shot classification pipeline
const classifier = await pipeline('zero-shot-classification', 'onnx-community/deberta-small-long-nli');
// Classify input text
const text = 'one day I will see the world';
const candidate_labels = ['travel', 'cooking', 'dancing'];
const output = await classifier(text, candidate_labels);
console.log(output);
// {
// sequence: 'one day I will see the world',
// labels: [ 'travel', 'dancing', 'cooking' ],
// scores: [ 0.8759302251623927, 0.10660000891028172, 0.017469765927325517 ]
// }
Note: Having a separate repo for ONNX weights is intended to be a temporary solution until WebML gains more traction. If you would like to make your models web-ready, we recommend converting to ONNX using 🤗 Optimum and structuring your repo like this one (with ONNX weights located in a subfolder named onnx
).
- Downloads last month
- 23
Inference API (serverless) does not yet support transformers.js models for this pipeline type.
Model tree for onnx-community/deberta-small-long-nli
Base model
microsoft/deberta-v3-small
Finetuned
tasksource/deberta-small-long-nli