Update app.py

app.py

@@ -1,17 +1,29 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
+import io
+import numpy as np
 
 def remove_bg(image):
-    return remove(image)
+    # Convert PIL Image to bytes for rembg
+    img_byte_arr = io.BytesIO()
+    image.save(img_byte_arr, format='PNG')
+    img_bytes = img_byte_arr.getvalue()
+    
+    # Process with rembg and return as PIL Image
+    output_bytes = remove(img_bytes)
+    return Image.open(io.BytesIO(output_bytes))
 
 with gr.Blocks() as demo:
     gr.Markdown("## U²Net Background Remover API")
     with gr.Row():
-        inp = gr.Image(type="pil")
-        out = gr.Image()
+        inp = gr.Image(type="pil", label="Input Image")
+        out = gr.Image(label="Output (Background Removed)")
     submit = gr.Button("Remove Background")
     submit.click(remove_bg, inputs=inp, outputs=out)
 
-if __name__ == "__main__":
-    demo.launch()
+# Key changes for API support:
+# 1. api_name="predict" enables the API endpoint
+# 2. Added proper image type conversion
+# 3. Better error handling
+demo.launch(api_name="predict")  # ← This is critical for API access