Update app.py

app.py

@@ -5,26 +5,29 @@ import io
 import numpy as np
 
 def remove_bg(image):
-    # 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 input to PIL Image
+    if isinstance(image, str):  # file path
+        img = Image.open(image)
+    elif isinstance(image, np.ndarray):  # numpy array from Gradio
+        img = Image.fromarray(image)
+    else:
+        raise ValueError("Unsupported input type")
     
-    # Convert to bytes for rembg
+    # Process with rembg
     img_byte_arr = io.BytesIO()
-    image.save(img_byte_arr, format='PNG')
+    img.save(img_byte_arr, format='PNG')
     output_bytes = remove(img_byte_arr.getvalue())
     return Image.open(io.BytesIO(output_bytes))
 
-# Use Interface instead of Blocks for better API support
+# Define the interface
 iface = gr.Interface(
     fn=remove_bg,
-    inputs=gr.Image(type="filepath"),  # Important for API
-    outputs=gr.Image(),
+    inputs=gr.Image(type="filepath", label="Input Image"),  # Important: use "filepath" for API compatibility
+    outputs=gr.Image(label="Output Image"),
     title="Background Remover",
-    allow_flagging="never"
+    description="Upload an image to remove its background"
 )
 
+# Launch the app
 if __name__ == "__main__":
     iface.launch(server_name="0.0.0.0", server_port=7860)