# Load model directly
from transformers import AutoImageProcessor, AutoModelForImageClassification
processor = AutoImageProcessor.from_pretrained("fxmarty/resnet-tiny-mnist")
model = AutoModelForImageClassification.from_pretrained("fxmarty/resnet-tiny-mnist")Quick Links
A small Resnet model for MNIST. Achieves 0.985 accuracy on the validation set.
- Downloads last month
- 197
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="fxmarty/resnet-tiny-mnist") pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")