can predict one image now
Now can predict a batch of images