dima806 commited on
Commit
341594f
·
verified ·
1 Parent(s): b2957e6

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -0
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.