problem modifying a function from handling one image to a batch of images

Question

I have created a function to receive an image and a category and identify it with percentages using pytorch pretrained models.

here it is:

def image_cl