more debugging

handler.py

@@ -1,12 +1,15 @@
 from io import BytesIO
 import base64
 import traceback
+import logging
 
 from PIL import Image
 import torch
 from transformers import CLIPProcessor, CLIPTextModel, CLIPVisionModelWithProjection
 
 device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')
+logger = logging.getLogger(__name__)
+logger.setLevel('INFO')
 
 class EndpointHandler():
     def __init__(self, path=""):
@@ -17,9 +20,11 @@ class EndpointHandler():
     def __call__(self, data):
         try:
 
+            logger.info('data is %s', data)
             text_input = None
             if isinstance(data, dict):
                 print('data is a dict: ', data)
+                logger.info('data is a dict %s', data)
                 inputs = data.pop("inputs", None)
                 text_input = inputs["text"] if "text" in inputs else None
                 image_data = BytesIO(base64.b64decode(inputs['image'])) if 'image' in inputs else None
@@ -39,5 +44,9 @@ class EndpointHandler():
                     return {'embeddings':self.image_model(**processor).image_embeds.tolist()[0]}
             else:
                 return {'embeddings':None}
-        except Exception:
-            return {'Error':traceback.format_exc()}
+        except Exception as ex:
+            logger.error('error doing request: %s', ex)
+            logger.exception(ex)
+            stack_info = traceback.format_exc()
+            logger.error('stack trace:\n%s',stack_info)
+            return {'Error':stack_info}