add print statement to handler

handler.py

@@ -33,10 +33,11 @@ class EndpointHandler:
 
     def __call__(self, data: dict[str, any]) -> dict[str, str]:
         base64_image = data.get('inputs')
-        image = self._decode_base64_image(base64_image)
-        output = self.generator(image)
-        output = output.squeeze(0)
-        output_image = transforms.ToPILImage()(output)
+        input_tensor = self._decode_base64_image(base64_image)
+        print('Input tensor shape: ' + str(input_tensor.shape))
+        output_tensor = self.generator(input_tensor)
+        output_tensor = output_tensor.squeeze(0)
+        output_image = transforms.ToPILImage()(output_tensor)
         output_image = output_image.convert('RGB')
         output_buffer = io.BytesIO()
         output_image.save(output_buffer, format="png")