README.md

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

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