Python Image Classification Code For Workers