Added the batch size for making predictions on