Update app.py

app.py

@@ -40,20 +40,17 @@ from rembg import remove
 def processRemove(image_file: Image.Image) -> Image.Image:
     if image_file is None:
         return None
-    
-    # Read the image data
-    if hasattr(image_file, 'read'):
-        input_data = image_file.read()
-    else:
-        with open(image_file, 'rb') as f:
-            input_data = f.read()
-    
-    # Remove background
+
+    # Chuyển ảnh PIL thành bytes
+    with BytesIO() as buffer:
+        image_file.save(buffer, format="PNG")
+        input_data = buffer.getvalue()
+
+    # Xóa nền
     output_data = remove(input_data)
-    
-    # Convert to PIL Image
-    output_image = Image.open(io.BytesIO(output_data))
-    return output_image
+
+    # Trả về ảnh PIL mới
+    return Image.open(BytesIO(output_data)).convert("RGBA")
 
 # --- Upscaling ---
 MAX_SEED = np.iinfo(np.int32).max