TianfuXinqu's picture
Model card
15f1b09 verified
|
Raw
History Blame Contribute Delete
500 Bytes
metadata
language: en
license: mit

Image Classifier

This model classifies images into 1000 ImageNet categories. It is based on a ResNet-50 architecture trained on ImageNet-1K.

How to use

from transformers import AutoModelForImageClassification, AutoImageProcessor

model = AutoModelForImageClassification.from_pretrained("TianfuXinqu/fetch-huggingface-9010-image-classifier")
processor = AutoImageProcessor.from_pretrained("TianfuXinqu/fetch-huggingface-9010-image-classifier")