So I am trying use a image recognition AI for a project. The module I using is the one on this website https://towardsdatascience.com/train-image-recognition-ai-with-5-lines-of-