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