from image_classification_model.train import train

if __name__ == "__main__":
    train()