Python knn image classifier