Update handler.py

handler.py

@@ -34,22 +34,37 @@ class EndpointHandler():
         #         image: await (await fetch('https://placekitten.com/300/300')).blob()
         #       }
         #     })
-        inputs = data.pop("inputs", data)
-        parameters = data.pop("parameters", {})
-        # if isinstance(inputs, Image.Image):
-        #     image = [inputs]
-        # else:
-        #     try:
-        #         imageBase64 = inputs["image"]
-        #         # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
-        #         image = Image.open(BytesIO(base64.b64decode(imageBase64)))
-    
-            # except:
-        image_url = inputs['image']
-        image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
-            
-        question = inputs["question"]
+        ###################
+        inputs = data.get("inputs")
+        imageBase64 = inputs.get("image")
+        # imageURL = inputs.get("image")
+        question = inputs.get("question")
+        # print(imageURL)
+        # print(text)
+        # image = Image.open(requests.get(imageBase64, stream=True).raw)
         
+        image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
+        ###################
+
+######################################
+
+        # inputs = data.pop("inputs", data)
+        # parameters = data.pop("parameters", {})
+        # # if isinstance(inputs, Image.Image):
+        # #     image = [inputs]
+        # # else:
+        # #     try:
+        # #         imageBase64 = inputs["image"]
+        # #         # image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[1].encode())))
+        # #         image = Image.open(BytesIO(base64.b64decode(imageBase64)))
+    
+        #     # except:
+        # image_url = inputs['image']
+        # image = Image.open(requests.get(image_url, stream=True).raw).convert('RGB')
+
+
+        # question = inputs["question"]
+######################################
         # data = data.pop("inputs", data)
         # data = data.pop("image", image)
 
@@ -71,7 +86,7 @@ class EndpointHandler():
 
         
         prompt = f"Question: {question}, Answer:"
-        processed = self.processor(images=image, text=prompt, return_tensors="pt").to(self.device)
+        processed = self.processor(images=image, text=prompt, return_tensors="pt")#.to(self.device)
 
         # answer = self._generate_answer(
         #     model_path, prompt, image,