# Base Model Classifier (ViT)

This model predicts the **base model used to generate an AI image**.

- Backbone: vit_base_patch16_224
- Classes: 31
- Task: image classification

## Usage

```python
from modeling import predict
print(predict("example.jpg"))