Multi Class Image Classification Python