Creating a Deep Learning Model for an Image Dataset