Change how we read the png image

handler.py

@@ -2,7 +2,6 @@ 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
@@ -35,8 +34,7 @@ class EndpointHandler():
         """
         prompt = data.pop("inputs", data)
         url = data.pop("url", data)
-        response = requests.get(url)
-        init_image = Image.open(BytesIO(response.content)).convert("RGB")
+        init_image = Image.open(url.convert("RGB")
         init_image.thumbnail((512, 512))
 
         params = data.pop("parameters", data)