# Load model directly
from transformers import AutoModelForImageClassification
model = AutoModelForImageClassification.from_pretrained("not-lain/deepfake", trust_remote_code=True, dtype="auto")Quick Links
original model repo :
📖 this is a cutomized version of the following model aaronespasa/deepfake-detection-resnetinceptionv1
how to use
from transformers import pipeline
pipe = pipeline(model="not-lain/deepfake",trust_remote_code=True)
pipe.predict("img_path.jpg")
>> {"confidences":confidences,"face_with_mask": face_with_mask}
dependencies
to install related dependencies simply use the command
!wget https://huggingface.co/not-lain/deepfake/resolve/main/requirements.txt && pip install -r requirements.txt
- Downloads last month
- 130
# Use a pipeline as a high-level helper from transformers import pipeline pipe = pipeline("image-classification", model="not-lain/deepfake", trust_remote_code=True) pipe("https://huggingface.co/datasets/huggingface/documentation-images/resolve/main/hub/parrots.png")