Update README.md

README.md

@@ -1,3 +1,20 @@
----
-license: cc-by-nc-4.0
----
+---
+license: cc-by-nc-4.0
+base_model:
+- Organika/sdxl-detector
+---
+# AI-image-detector
+The purpose of this model is to classify images as AI generated or Real.
+
+This model was created by fine-tuning the [Organika/sdxl-detector] on dataset of AI generated and real images from reddit, kaggle and real art from public domain with their text description. 
+Dataset was balanced to have similar number of real and generated images in each class (e.g. art, photos ...). 
+Art images from public domain were paired with generated equivalent created from their text descriptions with style transfer (sdxl with ip-adapter) from original piece. 
+The final dataset consisted of more than 50k images.
+
+The testing dataset consisted of 20% split of our base dataset and images outside the training domain from specific popular (as of 2024) image generation models.
+Finetuning vastly improved performance over Organika/sdxl-detector during testing, especially on images created by newer models. 
+
+The data used to fine-tune this model was scraped from image dedicated subreddits, some of which may be copyrighted. For this reason, this model should be considered appropriate only for non-comercial use only.
+
+
+