Update README.md
Browse files
README.md
CHANGED
|
@@ -1,5 +1,44 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
---
|
| 2 |
-
|
| 3 |
-
|
| 4 |
-
|
| 5 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
+
# 🔥 Deepfake Image Detection Model (Ensemble)
|
| 2 |
+
|
| 3 |
+
## 📌 Overview
|
| 4 |
+
This model detects whether an image is **REAL or FAKE (AI-generated)**.
|
| 5 |
+
|
| 6 |
+
It uses an **ensemble of 3 Vision Transformer (ViT/CLIP-based) models**:
|
| 7 |
+
- m1.safetensors
|
| 8 |
+
- m2.safetensors
|
| 9 |
+
- m3.safetensors
|
| 10 |
+
|
| 11 |
---
|
| 12 |
+
|
| 13 |
+
## 🧠 Model Details
|
| 14 |
+
|
| 15 |
+
| Model | Type |
|
| 16 |
+
|------|------|
|
| 17 |
+
| m1 | ViT / CLIP |
|
| 18 |
+
| m2 | ViT |
|
| 19 |
+
| m3 | Fine-tuned CLIP |
|
| 20 |
+
|
| 21 |
+
---
|
| 22 |
+
|
| 23 |
+
## 🎯 Task
|
| 24 |
+
Binary Classification:
|
| 25 |
+
- 0 → REAL
|
| 26 |
+
- 1 → FAKE
|
| 27 |
+
|
| 28 |
+
---
|
| 29 |
+
|
| 30 |
+
## 📥 Input
|
| 31 |
+
- Image (RGB)
|
| 32 |
+
- Size: 224 × 224
|
| 33 |
+
|
| 34 |
+
---
|
| 35 |
+
|
| 36 |
+
## 📤 Output
|
| 37 |
+
- Fake probability (0 to 1)
|
| 38 |
+
|
| 39 |
+
---
|
| 40 |
+
|
| 41 |
+
## ⚙️ Installation
|
| 42 |
+
|
| 43 |
+
```bash
|
| 44 |
+
pip install torch torchvision timm safetensors pillow
|