Spaces:
Sleeping
Sleeping
| 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) |