Hi, you can use any, but generator functions are mainly used only to handle large amount of data that didn’t fit in the memory, Since you’re going to predict for only one Image use the predict method. Anyway the predict function expects a batch of inputs instead of a single, so make sure you reshape the numpy array of the image to have one extra dimension (1 image inside 1batch).