from inference.models.vit.vit_classification import VitClassification