File size: 356 Bytes
aa94099 |
1 2 3 4 |
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) |