metadata
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
pip install torch torchvision timm safetensors pillow