Update app.py

app.py

@@ -5,11 +5,11 @@ import io
 import numpy as np
 
 def remove_bg(image):
-    # Convert input (could be numpy array from Gradio or file path from API)
-    if isinstance(image, np.ndarray):
-        image = Image.fromarray(image)
-    elif isinstance(image, str):
+    # Handle both file paths (API) and numpy arrays (UI)
+    if isinstance(image, str):  # API sends file path
         image = Image.open(image)
+    elif isinstance(image, np.ndarray):  # UI sends numpy array
+        image = Image.fromarray(image)
     
     # Convert to bytes for rembg
     img_byte_arr = io.BytesIO()
@@ -17,14 +17,14 @@ def remove_bg(image):
     output_bytes = remove(img_byte_arr.getvalue())
     return Image.open(io.BytesIO(output_bytes))
 
-# Create interface - this works better for APIs than Blocks
+# Use Interface instead of Blocks for better API support
 iface = gr.Interface(
     fn=remove_bg,
-    inputs=gr.Image(type="filepath", label="Input Image"),
-    outputs=gr.Image(label="Output"),
-    title="U²Net Background Remover API",
-    description="Upload an image to remove the background"
+    inputs=gr.Image(type="filepath"),  # Important for API
+    outputs=gr.Image(),
+    title="Background Remover",
+    allow_flagging="never"
 )
 
-# For Hugging Face Spaces, use this launch configuration
-iface.launch(server_name="0.0.0.0", server_port=7860)
+if __name__ == "__main__":
+    iface.launch(server_name="0.0.0.0", server_port=7860)