https://huggingface.co/distilbert-base-uncased-finetuned-sst-2-english with ONNX weights to be compatible with Transformers.js.
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
).
transformers = await import_transformers_js() pipeline = transformers.pipeline pipe = await pipeline('sentiment-analysis', 'osanseviero/distilbert-base-uncased-finetuned-quantized')
async def classify(text): return await pipe(text)
demo = gr.Interface(classify, "textbox", "json") demo.launch()
- Downloads last month
- 4
Inference API (serverless) does not yet support transformers.js models for this pipeline type.