Update app.py

app.py

@@ -4,6 +4,7 @@ from transparent_background import Remover
 from PIL import Image
 import numpy as np
 import io
+import tempfile
 
 @spaces.GPU
 def remove_background(image):
@@ -16,19 +17,18 @@ def remove_background(image):
     else:
         raise TypeError("Unsupported image type")
     
-    # 确保输出图片的背景是透明的，并以PNG格式保存
-    output_bytes = io.BytesIO()
-    output.save(output_bytes, format="PNG")
-    output_bytes.seek(0)
-    output = Image.open(output_bytes)
+    # 保存PNG图片到临时文件
+    temp_file = tempfile.NamedTemporaryFile(delete=False, suffix=".png")
+    output.save(temp_file, format="PNG")
+    temp_file.close()
 
-    return output
+    return temp_file.name
 
 # 界面设置
 iface = gr.Interface(
     fn=remove_background,
     inputs=gr.Image(label="Upload Image"),
-    outputs=gr.Image(label="Output Image", type="pil"),
+    outputs=gr.Image(label="Output Image", type="filepath"),
     title="AI Background Remover - Automatically Remove Image Backgrounds",
     description="Upload an image and our AI-powered background remover will automatically make the background transparent. Perfect for various needs requiring transparent images.",
     article="""