SadraCoding commited on
Commit
83c560d
·
verified ·
1 Parent(s): 9164a4d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +12 -1
README.md CHANGED
@@ -14,7 +14,18 @@ As generative AI (like SDXL, DALL·E, and Midjourney) becomes more accessible, t
14
 
15
  ## 🚀 Model Overview
16
 
17
- **SDXL-Deepfake-Detector** is a fine-tuned vision transformer that classifies human faces as **AI-Generated (0)** or **Real (1)**. Trained on the [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) dataset.
 
 
 
 
 
 
 
 
 
 
 
18
 
19
  ### ✅ Key Highlights
20
  - **Architecture**: Fine-tuned Vision Transformer (ViT) via Hugging Face `transformers`
 
14
 
15
  ## 🚀 Model Overview
16
 
17
+ **SDXL-Deepfake-Detector** is a fine-tuned vision transformer that classifies human faces as **AI-Generated (0)** or **Real (1)**.
18
+
19
+ ## 🧠 Training Approach
20
+
21
+ This model was obtained by **fine-tuning** the [`Organika/sdxl-detector`](https://huggingface.co/Organika/sdxl-detector) — a vision transformer pre-trained specifically to detect SDXL-generated faces — on the [140k Real and Fake Faces](https://www.kaggle.com/datasets/xhlulu/140k-real-and-fake-faces) dataset.
22
+
23
+ This approach leverages:
24
+ - Prior knowledge of SDXL artifacts from the base model
25
+ - Broader generalization from a large-scale real/fake face dataset
26
+ - Efficient training on limited hardware (single RTX 3060)
27
+
28
+ The result is a lightweight, high-accuracy detector optimized for **both SDXL and general diffusion-based deepfakes**.
29
 
30
  ### ✅ Key Highlights
31
  - **Architecture**: Fine-tuned Vision Transformer (ViT) via Hugging Face `transformers`