from .vit_model import create_name_vit
from .classification import ClassificationModel