File size: 203 Bytes
d3e1b7a
 
 
 
1
2
3
4
5

from transformers import AutoModelForImageClassification
model = AutoModelForImageClassification.from_pretrained("prithivMLmods/Deep-Fake-Detector-v2-Model")
print("LABELS:", model.config.id2label)