dima806
/

ai_vs_human_generated_image_detection

Image Classification

Model card Files Files and versions Community

dima806 commited on Jan 25

Commit

b2957e6

·

verified ·

1 Parent(s): 2af5ab4

Update README.md

Files changed (1) hide show

README.md +7 -7

README.md CHANGED Viewed

@@ -8,21 +8,21 @@ base_model:
 pipeline_tag: image-classification
 library_name: transformers
 ---
-Predicts with about 97% accuracy whether an attached image is AI-generated.
 See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for details.
-![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/RBfE9AIiEMlriYujz2FJQ.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
 ```

 pipeline_tag: image-classification
 library_name: transformers
 ---
+Predicts with about 98% accuracy whether an attached image is AI-generated.
 See https://www.kaggle.com/code/dima806/ai-vs-human-generated-images-prediction-vit for details.
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6449300e3adf50d864095b90/vaVofgP3NO3ybwG0GSbC3.png)
 ```
 Classification report:
               precision    recall  f1-score   support
+       human     0.9655    0.9930    0.9790      3998
+AI-generated     0.9928    0.9645    0.9784      3997
+    accuracy                         0.9787      7995
+   macro avg     0.9791    0.9787    0.9787      7995
+weighted avg     0.9791    0.9787    0.9787      7995
 ```