binary

handler.py

@@ -79,8 +79,11 @@ class EndpointHandler():
         elif input_type == "processed_images":
             try:
                 binary = base64.b64decode(inputs)
-                batch = torch.load(io.BytesIO(inputs), map_location=self.model.device)
-                print(batch)
+                print("decoded binary")
+                buffer = io.BytesIO(binary)
+                print("buffered")
+                batch = torch.load(buffer, map_location=self.model.device)
+                print("loaded as torch")
             except Exception as e:
                 return {"error": f"Error processing preprocessed images: {str(e)}"}