Update handler.py

handler.py

@@ -2,6 +2,7 @@ from transformers import AutoModelForCausalLM, AutoProcessor
 from PIL import Image
 import requests
 import torch
+import io
 
 class EndpointHandler:
     def __init__(self, model_dir):
@@ -17,7 +18,7 @@ class EndpointHandler:
                 return {"error": "Missing URL"}
             
             response = requests.get(url, verify=False)
-            image = Image.open(response.raw).convert("RGB")
+            image = Image.open(io.BytesIO(response.content)).convert("RGB")
             
             inputs = self.processor(
                 text="<MORE_DETAILED_CAPTION>",