| --- |
| --- |
| license: mit |
| language: en |
| pipeline_tag: image-classification |
| tags: |
| - deepfake |
| - computer-vision |
| - image-classification |
| - pytorch |
| --- |
| --- |
| # 🔥 Deepfake Image Detection Model (Ensemble) |
|
|
| ## 📌 Overview |
| This model detects whether an image is **REAL or FAKE (AI-generated)**. |
|
|
| It uses an **ensemble of 2 Vision Transformer (ViT & EfficientNet-B4) models**: |
|
|
| - model.safetensors |
|
|
| --- |
|
|
| ## 🧠 Model Details |
|
|
| model safetensor |
| --- |
|
|
| ## 🎯 Task |
| Binary Classification: |
| - 0 → REAL |
| - 1 → FAKE |
|
|
| --- |
|
|
| ## 📥 Input |
| - Image (RGB) |
| - Size: 224 × 224 |
|
|
| --- |
|
|
| ## 📤 Output |
| - Fake probability (0 to 1) |
|
|
| --- |
|
|
| ## ⚙️ Installation |
|
|
| ```bash |
| pip install torch torchvision timm safetensors pillow |
