import argparse
from models import ClassificationModel
import cv2 as cv

def main(args):
    cls_model = ClassificationModel()
    print(cls_model.predict(cv.imread(args.image_path)))
    
if __name__ == "__main__":
    parser = argparse.ArgumentParser()
    parser.add_argument("--image_path", type=str, required=True)
    args = parser.parse_args()
    main(args)