Add conversion of extracted image to PIL format in remove_background function in app.py. This ensures compatibility with subsequent image processing steps when the output is a NumPy array.

app.py

@@ -74,6 +74,9 @@ def remove_background(
     with tempfile.NamedTemporaryFile(suffix=".png", delete=True) as temp:
         image.save(temp.name)
         extracted_img, mask = dis_remove_background(DIS_ONNX_MODEL_PATH, temp.name)
+    # Convert to PIL Image if needed
+    if isinstance(extracted_img, np.ndarray):
+        extracted_img = Image.fromarray(extracted_img)
     return extracted_img
 
 def do_resize_content(original_image: Image, scale_rate):