Update handler.py

handler.py

@@ -1,6 +1,8 @@
 from typing import Dict, List, Any
 import base64
 
+import logging
+
 import tensorflow as tf
 from tensorflow import keras
 from keras_cv.models.generative.stable_diffusion.text_encoder import TextEncoder
@@ -73,4 +75,11 @@ class EndpointHandler():
         unconditional_context_b64 = base64.b64encode(unconditional_context.numpy().tobytes())
         unconditional_context_b64str = unconditional_context_b64.decode()        
 
+        print("1")
+        print(context_b64)
+        logging.info("2")
+        logging.info(context_b64)
+        logging.warning("3")
+        logging.warning(context_b64)
+        
         return {"context_b64str": context_b64str, "unconditional_context_b64str": unconditional_context_b64str}