dima806 commited on
Commit
b258c3c
·
verified ·
1 Parent(s): f34fc3b

Create README.md

Browse files

Predicts with about 97% accuracy whether an attached image is a human made or AI generated.

See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for more details.

![image.png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/w7Gdjewo-TJn0vkfvmrKx.png)
```
Classification report:

precision recall f1-score support

human 0.9606 0.9849 0.9726 15990
AI-generated 0.9845 0.9596 0.9719 15990

accuracy 0.9723 31980
macro avg 0.9726 0.9723 0.9723 31980
weighted avg 0.9726 0.9723 0.9723 31980
```

Files changed (1) hide show
  1. README.md +10 -0
README.md ADDED
@@ -0,0 +1,10 @@
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ metrics:
4
+ - accuracy
5
+ - f1
6
+ base_model:
7
+ - google/vit-base-patch16-224-in21k
8
+ pipeline_tag: image-classification
9
+ library_name: transformers
10
+ ---