continue debugging adventures

handler.py

@@ -1,3 +1,4 @@
+import sys
 from io import BytesIO
 import base64
 
@@ -15,19 +16,19 @@ class EndpointHandler():
 
     def __call__(self, data):
         inputs = data.pop("inputs", None)
-        print(inputs)
+        print(inputs, file=sys.stderr)
         text_input = inputs["text"] if "text" in inputs else None
         image_input = inputs["image"] if "image" in inputs else None
 
         if text_input:
-            print("in text mode")
-            print(text_input)
+            print("in text mode", file=sys.stderr)
+            print(text_input, file=sys.stderr)
             processor = self.processor(text=text_input, return_tensors="pt", padding=True)
             with torch.no_grad():
                 return self.text_model(**processor).pooler_output.tolist()
         elif image_input:
-            print("in image mode")
-            print(image_input)
+            print("in image mode", file=sys.stderr)
+            print(image_input, file=sys.stderr)
             image = Image.open(BytesIO(base64.b64decode(image_input)))
             processor = self.processor(images=image, return_tensors="pt")
             with torch.no_grad():