Update README.md
Browse files
README.md
CHANGED
|
@@ -8,6 +8,16 @@ base_model:
|
|
| 8 |
pipeline_tag: image-classification
|
| 9 |
library_name: transformers
|
| 10 |
---
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 11 |
Predicts with about 98% accuracy whether an attached image is AI-generated.
|
| 12 |
|
| 13 |
See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for details.
|
|
|
|
| 8 |
pipeline_tag: image-classification
|
| 9 |
library_name: transformers
|
| 10 |
---
|
| 11 |
+
**Note to users who want to use this model in production**
|
| 12 |
+
|
| 13 |
+
Beware that this model is trained on a dataset collected **about 1 year ago**.
|
| 14 |
+
Since then, there is a remarkable progress in generating deepfake images with common AI tools, resulting in a significant concept drift.
|
| 15 |
+
To mitigate that, I urge you to retrain the model using the latest available labeled data.
|
| 16 |
+
As a quick-fix approach, simple reducing the threshold (say from default 0.5 to 0.1 or even 0.01) of labelling image as a fake may suffice.
|
| 17 |
+
However, you will do that at your own risk, and retraining the model is the better way of handling the concept drift.
|
| 18 |
+
|
| 19 |
+
=======
|
| 20 |
+
|
| 21 |
Predicts with about 98% accuracy whether an attached image is AI-generated.
|
| 22 |
|
| 23 |
See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for details.
|