binary

handler.py

@@ -78,12 +78,8 @@ class EndpointHandler():
 
         elif input_type == "processed_images":
             try:
-                binary = base64.b64decode(inputs)
-                print("decoded binary")
-                buffer = io.BytesIO(binary)
-                print("buffered")
+                buffer = io.BytesIO(base64.b64decode(inputs))
                 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)}"}