---
# For reference on model card metadata, see the spec: https://github.com/huggingface/hub-docs/blob/main/modelcard.md?plain=1
# Doc / guide: https://huggingface.co/docs/hub/model-cards
{}
---

# Model Card for VGG19

VGG19BN Model from torchvision.models

Used in 'aiornot' classification competition on HuggingFace. Performs binary classification to detect whether the image is AI-generated or not.

## Model Details

### How to Get Started with the Model

Use the code below to get started with the model.

    ckpt = torch.load('weights.ckpt', map_location=device)
    weights = ckpt['model']
    model = torchvision.models.vgg19_bn().to(device)
    model.load_state_dict(weights)