Update handler.py

handler.py

@@ -8,11 +8,7 @@ from tensorflow import keras
 from keras_cv.models.generative.stable_diffusion.decoder import Decoder
 
 class EndpointHandler():
-    def __init__(self, path=""):
-        gpus = tf.config.list_physical_devices(device_type = 'GPU')
-        for gpu in gpus:
-            tf.config.experimental.set_memory_growth(gpu, True)
-        
+    def __init__(self, path=""):        
         img_height = 512
         img_width = 512
         img_height = round(img_height / 128) * 128
@@ -27,7 +23,7 @@ class EndpointHandler():
 
     def __call__(self, data: Dict[str, Any]) -> str:
         # get inputs 
-        logging.info(data)
+        logging.warning(data)
         latent = data.pop("inputs", data)
         batch_size = data.pop("batch_size", 1)