I have a previous story about training an image classifier… towards the end there’s my code to use the trained model on other photos:

https://towardsdatascience.com/how-to-train-an-image-classifier-in-pytorch-and-use-it-to-perform-basic-inference-on-single-images-99465a1e9bf5