Simma7 commited on
Commit
68e4a3d
·
verified ·
1 Parent(s): cac5c9d

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +43 -4
README.md CHANGED
@@ -1,5 +1,44 @@
 
 
 
 
 
 
 
 
 
 
1
  ---
2
- license: mit
3
- language:
4
- - en
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