# pip install -U transformers accelerate
# Load model directly
from transformers import AutoModel
model = AutoModel.from_pretrained("Roy229/nml7324-image-classifier", device_map="auto")Quick Links
Image Classifier
Classifies product photos into 1,200 fine-grained categories for the retail catalogue. Trained on an internal dataset of e-commerce images and deployed behind the catalogue search service. It is the highest-traffic model in the registry with roughly eight million inference requests per week.
- Downloads last month
- 4
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="Roy229/nml7324-image-classifier") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")