Update handler.py

handler.py

@@ -21,22 +21,15 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> Dict[str, Any]:
         logging.error(f"----------This is an error message")
-        #logging.critical('-------------------This is a critical message')
-        print("000000--",type(data))
-        #logging.warning('--------This is a warning message')
-        print("input data is here------------",data)
         input_data = data.get("inputs", {})
-        logging.warning('------input_data--This is a warning message', input_data)
-        print("input data is here-2-----------",type(input_data))
+        logging.warning(f"------input_data-- {str(input_data)}")
         encoded_images = input_data.get("images")
-        logging.warning(f"---encoded_images-----This is a warning message {str(encoded_images)}")
-        print("input encoded_images is here------------",type(encoded_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)}")
             return {"captions": [], "error": "No images provided"} 
-
-        #texts = input_data.get("texts", ["a photography of"] * len(encoded_images)) 
-
         try:
+            logging.warning(f"---encoded_images-- provided in try block--- {str(encoded_images)}")
             byteImgIO = io.BytesIO()
             
             byteImg = Image.open(encoded_images[0])
@@ -48,7 +41,7 @@ class EndpointHandler():
             # Non test code
             dataBytesIO = io.BytesIO(byteImg)
             raw_images =[Image.open(dataBytesIO)]
-            logging.warning(f"----raw_images----This is a warning message {str(raw_images)}")
+            logging.warning(f"----raw_images----0--- {str(raw_images)}")
             # Check if any images were successfully decoded
             if not raw_images:
                 print("No valid images found.")
@@ -63,7 +56,7 @@ class EndpointHandler():
                 out = self.model.generate(**processed_inputs)
 
             captions = self.processor.batch_decode(out, skip_special_tokens=True)
-            logging.warning(f"----captions----This is a warning message {str(captions)}")
+            logging.warning(f"----captions---- {str(captions)}")
             print("caption is here-------",captions)
             return {"captions": captions}
         except Exception as e: