Update u2net_utils.py

u2net_utils.py

@@ -1,6 +1,7 @@
 import torch
 import torchvision.transforms as transforms
 from PIL import Image
+from io import BytesIO
 import numpy as np
 import os
 import requests
@@ -67,4 +68,7 @@ def remove_background(image: Image.Image) -> Image.Image:
     new_image = Image.new("RGBA", original_size)
     new_image.paste(image, (0, 0), mask=mask)
 
-    return new_image
+    buffer = BytesIO()
+new_image.save(buffer, format="PNG", optimize=False, compress_level=1)
+buffer.seek(0)
+return buffer