## pretrained_model for image classification used convnext-tiny-224