Image Classification Code In Python For Prime