Upload folder using huggingface_hub

handler.py

@@ -39,7 +39,7 @@ class EndpointHandler:
                 else:
                     image = self._decode_image(item)
                     # print(f"Processing image: {image.size} {image.mode}")
-                    inputs = self.processor(images=image, return_tensors="pt").to(self.device)
+                    inputs = self.processor(images=[image], return_tensors="pt").to(self.device)
                     with torch.no_grad():
                         features = self.model.get_image_features(**inputs)
                     results.append(features[0].cpu().tolist())