Tensorflow Datasets Cats Vs Dogs Game