image

handler.py

@@ -2,7 +2,7 @@ from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoToken
 import torch
 from PIL import Image
 from typing import Dict, List, Any
-import urllib.request
+import requests
 
 
 class EndpointHandler():
@@ -36,7 +36,11 @@ class EndpointHandler():
         image_paths = data.pop("image_paths", data)
         images = []
         for image_path in image_paths:
-            urllib.request.urlretrieve(image_path, "temp")
+            response = requests.get(image_path)
+            response.raise_for_status()  # Raise an exception if the request failed
+
+            with open("temp", "wb") as f:
+                f.write(response.content)
             i_image = Image.open("temp")
             if i_image.mode != "RGB":
                 i_image = i_image.convert(mode="RGB")

requirements.txt

@@ -1,4 +1,4 @@
 torch
 transformers
 Pillow
-urllib3
+requests