AI_Detector_App / models /image_model.py
ash12321's picture
Create models/image_model.py
050cc37 verified
raw
history blame contribute delete
250 Bytes
from transformers import pipeline
# Public, stable image classifier (generic). You can swap to a deepfake-specific model later.
image_model = pipeline(
task="image-classification",
model="google/vit-base-patch16-224",
device=-1 # CPU
)