File size: 356 Bytes
d3e1b7a | 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}")
|