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