Spaces:
Sleeping
Sleeping
File size: 314 Bytes
9d06c04 |
1 2 3 4 |
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) |