File size: 777 Bytes
9e2b557 106ab9e 6193dcc 106ab9e 569e9f0 106ab9e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 |
---
license: apache-2.0
datasets:
- competitions/aiornot
language:
- en
metrics:
- accuracy
tags:
- generative ai
- classification
---
Classification model used to classify real images and AI generated images.\
The model used is swin-tiny-patch4-window7-224 finetued on aiornot dataset.\
To use the model
```
import torch
from transformers import AutoFeatureExtractor, AutoModelForImageClassification
labels = ["Real", "AI"]
feature_extractor = AutoFeatureExtractor.from_pretrained("Nahrawy/AIorNot")
model = AutoModelForImageClassification.from_pretrained("Nahrawy/AIorNot")
input = feature_extractor(image, return_tensors="pt")
with torch.no_grad():
outputs = model(**input)
logits = outputs.logits
prediction = logits.argmax(-1).item()
label = labels[prediction]
``` |