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: I would like something like this returned: