Python Image Classification Code 8017