Hugging Face's logo

SupremoUGH
/
image-classification-model

Image Classification
Transformers
Safetensors
English
vit
vision
Eval Results
Model card Files Community
image-classification-model
File size: 1,194 Bytes 
ab8b628
56e21dd
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
ab8b628
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
 
---
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:

```bash
pip3 install requirements.txt
```

### Run unit tests
```bash
python3 -m unittest discover -s tests
```