File size: 257 Bytes
768ca02
10f6d20
768ca02
10f6d20
768ca02
 
 
2a1ca13
768ca02
2a1ca13
 
768ca02
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
# 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"))