ai-detect / debug_load.py
maddyrox's picture
Upload 23 files
d3e1b7a verified
raw
history blame contribute delete
356 Bytes
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}")