POC: Support batch prediction

To speed up the processing on Arkindex, we could predict images by batch. So far, the prediction function allows to predict only one image at a time.