Computer Vision model for finding if an image is of a cat or dog