Update README.md

README.md

@@ -9,4 +9,22 @@ metrics:
 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
+```
+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] 
+```