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