Update handler.py

handler.py

@@ -19,7 +19,7 @@ class EndpointHandler():
         self.processor = AutoProcessor.from_pretrained(self.model_name)
         self.model = Blip2ForConditionalGeneration.from_pretrained(self.model_name,
                                                               device_map="auto", 
-                                                             ).to(self.device)
+                                                             )#.to(self.device)
 
     def __call__(self, data: Dict[str, Any]) -> List[Dict[str, Any]]:
         """
@@ -44,7 +44,13 @@ class EndpointHandler():
         # print(imageURL)
         # print(text)
         # image = Image.open(requests.get(imageBase64, stream=True).raw)
-        
+        import base64
+        from PIL import Image
+        from io image BytesIO
+        import matplotlib.pyplot as plt
+        #try2
+        # image = Image.open(BytesIO(base64.b64decode(imageBase64)))
+        #try1
         image = Image.open(BytesIO(base64.b64decode(imageBase64.split(",")[0].encode())))
         ###################