Added support for the classification of a single image at a time, instead of using batches of images