File size: 178 Bytes
05957f7
 
 
 
1
2
3
4

# Load model directly
from transformers import AutoModelForImageClassification
model = AutoModelForImageClassification.from_pretrained("Diego192/prueba", trust_remote_code=True)