added logger

handler.py

@@ -95,11 +95,10 @@ class EndpointHandler():
         )
         
     def __call__(self, data):
-        # Now you can use logger.info, logger.warning, etc.
-        logger.info("This is an info message")
-        logger.warning("This is a warning message")
 
-        print("Start generating image...")
+        logger.info("Starting image generation...")
+        print("Starting image generation...")
+        
         def create_image(
             image_pil,
             input_image,
@@ -114,6 +113,7 @@ class EndpointHandler():
             neg_content_prompt=None,
             neg_content_scale=0,
         ):
+            logger.info("Creating seed...")
             print("Creating seed")
             seed = random.randint(0, MAX_SEED) if seed == -1 else seed
             # if target == "Load original IP-Adapter":
@@ -222,7 +222,8 @@ class EndpointHandler():
             return input_image
 
 
-        print("functions loaded...")
+        logger.info("Functions loaded...")
+        print("Functions loaded...")
         
         style_image_url = "https://huggingface.co/spaces/radames/InstantStyle-SDXL-Lightning/resolve/main/assets/0.jpg"
         response = requests.get(style_image_url)
@@ -233,11 +234,13 @@ class EndpointHandler():
         prompt =  "A Art Deco style artwork of tennis court with tennis balls and rackets, a modern car, (featuring Notre Dame Cathedral:1.5)"
         scale =2.0
         control_scale =0.0
-
+        
+        logger.info("Parameters loaded...")
         print("Parameters loaded...")
 
         try:
-            
+
+            logger.info("Generating image...")
             print("Generating image...")
                         
             return create_image(
@@ -255,5 +258,6 @@ class EndpointHandler():
             neg_content_scale=0,
             )
         except Exception as e:
+            logger.info(f"123 Error during image generation: {str(e)}")
             print(f"123 Error during image generation: {str(e)}")
             return None