Add the get back in

handler.py

@@ -2,6 +2,7 @@ from typing import  Dict, List, Any
 import torch
 import requests
 from PIL import Image
+from io import BytesIO
 from diffusers import StableDiffusionPipeline, StableDiffusionImg2ImgPipeline, DDIMScheduler
 
 # set device
@@ -34,7 +35,8 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", data)
         url = data.pop("url", data)
-        init_image = Image.open(url).convert("RGB")
+        response = requests.get(url)
+        init_image = Image.open(BytesIO(response.content)).convert("RGB")
         init_image.thumbnail((512, 512))
 
         params = data.pop("parameters", data)