Revert "rembg b - return a stream of RGB24 instead of a mess of conse…

…cutive pngs"
danielgatis · Jan 30, 2025 · b9634aa · b9634aa
1 parent 1f98d03
commit b9634aa
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/rembg/commands/b_command.py b/rembg/commands/b_command.py
@@ -135,7 +135,9 @@ def b_command(
             os.makedirs(output_dir, exist_ok=True)
 
     def img_to_byte_array(img: PILImage) -> bytes:
-        return img.convert('RGB').tobytes("raw", "RGB")
+        buff = io.BytesIO()
+        img.save(buff, format="PNG")
+        return buff.getvalue()
 
     async def connect_stdin_stdout():
         loop = asyncio.get_event_loop()