Python Image Classification Code 8871