Update handler.py

handler.py

@@ -21,7 +21,8 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         logging.error(f"----------This is an error message {str(data)}")
-        raw_images = data.get("inputs", {})
+        input_data = data.get("inputs", {})
+        logging.warning(f"------input_data-- {str(input_data)}")
     #     logging.warning(f"------input_data-- {str(input_data)}")
     #     encoded_images = input_data.get("images")
     #     # Convert image to bytes
@@ -50,6 +51,8 @@ class EndpointHandler():
             # raw_images =[Image.open(dataBytesIO)]
             # logging.warning(f"----raw_images----0--- {str(raw_images)}")
             # Check if any images were successfully decoded
+            raw_images = input_data
+            logging.warning(f"---raw_images-raw_images----------- {str(raw_images)}")
             if not raw_images:
                 print("No valid images found.")
             processed_inputs = [