Update handler.py

handler.py

@@ -1,6 +1,7 @@
 from typing import Dict, List, Any
 
 import base64
+import logging
 import numpy as np
 from tensorflow import keras
 from keras_cv.models.generative.stable_diffusion.decoder import Decoder
@@ -21,7 +22,7 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> str:
         # get inputs 
-        print(data)
+        logging.info(data)
         latent = data.pop("inputs", data)
         batch_size = data.pop("batch_size", 1)