Modelpublic

shivani1511/deepfake-image-detector-v2

sourceHugging Faceupdated 1y agoView on Hugging Face

0likes58downloads

Model Card

license: mit task: image-classification tags:

—deepfake-detection

Deepfake Image Detector (v2)

Model Details

—Base Model: google/vit-base-patch16-224
—Classes: 0_real, 1_fake
—Test Accuracy: 99.28%
—Training Samples: 100,000
—Validation Samples: 20,000

Usage

python

from transformers import ViTForImageClassification, ViTFeatureExtractor

model = ViTForImageClassification.from_pretrained('shivani1511/deepfake-image-detector-v2')
feature_extractor = ViTFeatureExtractor.from_pretrained('shivani1511/deepfake-image-detector-v2')