CoolFace
Modelpublic

juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized

sourceHugging Faceupdated 6mo agoView on Hugging Face
0likes23downloads
Model Card

DistilBERT Amazon Shoe Reviews — ONNX Optimized

Graph-optimized ONNX export of juliensimon/distilbert-amazon-shoe-reviews for faster CPU inference using Optimum.

Video walkthrough: Accelerate Transformer inference on CPU with Optimum and ONNX

Model Details

DetailValue
Base modeljuliensimon/distilbert-amazon-shoe-reviews
TaskStar rating prediction (1–5 stars) from shoe reviews
FormatONNX with graph optimizations (operator fusion, constant folding)

Usage

python
from optimum.onnxruntime import ORTModelForSequenceClassification
from transformers import AutoTokenizer, pipeline

model = ORTModelForSequenceClassification.from_pretrained("juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized")
tokenizer = AutoTokenizer.from_pretrained("juliensimon/distilbert-amazon-shoe-reviews-onnx-optimized")
classifier = pipeline("text-classification", model=model, tokenizer=tokenizer)

classifier("These shoes are incredibly comfortable!")

Related Models