external url

handler.py

@@ -2,6 +2,8 @@ from transformers import VisionEncoderDecoderModel, ViTImageProcessor, AutoToken
 import torch
 from PIL import Image
 from typing import Dict, List, Any
+import urllib.request
+
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -34,7 +36,8 @@ class EndpointHandler():
         image_paths = data.pop("image_paths", data)
         images = []
         for image_path in image_paths:
-            i_image = Image.open(image_path)
+            urllib.request.urlretrieve(image_path, "temp")
+            i_image = Image.open("temp")
             if i_image.mode != "RGB":
                 i_image = i_image.convert(mode="RGB")
 

requirements.txt

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