from utils.ml import train_model, save_model, load_model, predict

if __name__ == "__main__":
    model = train_model()
    did_save = save_model(model=model)
    if did_save:
        print("Model saved")
    loaded_model = load_model()
    pred = predict(
        model=loaded_model,
        sepal_length=4.2,
        sepal_width=3.1,
        petal_length=3.4,
        petal_width=4.1
    )
    print(pred)