| # 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")) | |