Batch infererence with images

Hi. Is it possible to provide inference of multi labels on a batch of images, including the image itself? get_preds(…) seems to only return the labels, not the associated images:
image
I would like something like this returned: