---
---
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