Update handler.py

handler.py

@@ -23,16 +23,16 @@ class EndpointHandler():
         logging.error(f"----------This is an error message {str(data)}")
         input_data = data.get("inputs", {})
         logging.warning(f"------input_data-- {str(input_data)}")
-        encoded_images = input_data.get("images")
-        logging.warning(f"---encoded_images----- {str(encoded_images)}")
-        if not encoded_images:
-            logging.warning(f"---encoded_images--not provided in if block--- {str(encoded_images)}")
+       # encoded_images = input_data.get("images")
+        #logging.warning(f"---encoded_images----- {str(encoded_images)}")
+        if not input_data:
+            #logging.warning(f"---encoded_images--not provided in if block--- {str(encoded_images)}")
             return {"captions": [], "error": "No images provided"} 
         try:
-            logging.warning(f"---encoded_images-- provided in try block--- {str(encoded_images)}")
+            logging.warning(f"---encoded_images-- provided in try block--- {str(input_data)}")
             byteImgIO = io.BytesIO()
             
-            byteImg = Image.open(encoded_images[0])
+            byteImg = Image.open(input_data)
             byteImg.save(byteImgIO, "PNG")
             byteImgIO.seek(0)
             byteImg = byteImgIO.read()