from transformers import AutoModelForImageClassification, AutoImageProcessor m = AutoModelForImageClassification.from_pretrained("dshi01/convnext-tiny-224-7clss") p = AutoImageProcessor.from_pretrained("dshi01/convnext-tiny-224-7clss") print(m.config.model_type, m.config.num_labels) # expect convnext, 7 (if set)