Update handler.py

handler.py

@@ -2,9 +2,12 @@ from typing import Dict, List, Any
 from transformers import AutoImageProcessor, Swin2SRForImageSuperResolution
 import torch
 import base64
+import logging
 from PIL import Image
 from io import BytesIO
 
+logger = logging.getLogger()
+logger.setLevel(logging.DEBUG)
 
 # check for GPU
 device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
@@ -19,8 +22,10 @@ class EndpointHandler:
         self.model.to(device)
 
     def __call__(self, data: Any) -> List[List[Dict[str, float]]]:
-        
+        logger.info(f"data: {data}")
+        logger.info(f"data type: {type(data)}")
         image = Image.open(BytesIO(base64.b64decode(data)))
+        logger.info("image read")
         inputs = self.processor(image, return_tensors="pt")
         outputs = self.model(**inputs)