Create README.md

README.md

@@ -0,0 +1,24 @@
+---
+# 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('last.ckpt', map_location=device)
+    weights = ckpt['model']
+    model = torchvision.models.vgg19_bn().to(device)
+    model.load_state_dict(weights)
+
+