Refactor remove_background function in app.py to save images to a temporary file before processing. This change improves compatibility with the dis_remove_background method by ensuring the image is correctly handled as a file path.

app.py

@@ -15,6 +15,7 @@ import os
 import json
 import argparse
 import requests
+import tempfile
 
 from model import CRM
 from inference import generate3d
@@ -69,7 +70,10 @@ def remove_background(
 ) -> PIL.Image.Image:
     # Ensure the ONNX model exists (download if needed)
     ensure_dis_onnx_model()
-    extracted_img, mask = dis_remove_background(DIS_ONNX_MODEL_PATH, image)
+    # Save PIL image to a temporary file
+    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)
     return extracted_img
 
 def do_resize_content(original_image: Image, scale_rate):