Hugging Face's logo

SupremoUGH
/
image-classification-model

Image Classification
Transformers
Safetensors
English
vit
vision
Eval Results
Inference Endpoints
Model card Files Community
image-classification-model / README.md
SupremoUGH's picture
SupremoUGH
update README
56e21dd unverified
preview code
|
raw
history blame contribute delete
1.19 kB
metadata
language: en
tags:
  - image-classification
  - vision
model-index:
  - name: ViT Image Classification Model
    sources:
      - https://huggingface.co/SupremoUGH/image-classification-model
    results:
      - task:
          name: image-classification
          type: image-classification
        metrics:
          - name: Accuracy
            value: 98.0%
            type: float
library_name: transformers
license: mit

Image Classification Model (ViT)

This is an image classification model based on Vision Transformer (ViT), fine-tuned on the MNIST dataset. The model is designed to classify images into one of 10 possible classes (digits 0-9). The code is compatible with Hugging Face's inference providers and can be easily deployed.

Model Details

  • Model Type: Vision Transformer (ViT)
  • Base Model: google/vit-base-patch16-224
  • Task: Image Classification
  • Dataset: MNIST (handwritten digits)
  • Labels: 10 classes (0-9)

How to Use

Install Requirements

Make sure you have the following dependencies installed:

pip3 install requirements.txt

Run unit tests 

python3 -m unittest discover -s tests