File size: 356 Bytes
6361699
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10

from transformers import AutoModelForImageClassification, AutoImageProcessor
try:
    print("Loading EfficientNet...")
    m = AutoModelForImageClassification.from_pretrained("Dafilab/ai-image-detector")
    p = AutoImageProcessor.from_pretrained("Dafilab/ai-image-detector")
    print("Success")
except Exception as e:
    print(f"FAILED: {e}")