Update app.py

app.py

@@ -2,20 +2,22 @@
 import gradio as gr
 from rembg import remove
 from PIL import Image
-import io
+import numpy as np
 
-# Function to remove background
 def remove_background(image):
-    # Convert uploaded image to PIL
+    """
+    Remove background from an uploaded image using rembg CPU backend.
+    Returns the processed image with transparency.
+    """
+    # Ensure the image is RGBA
     pil_image = Image.fromarray(image).convert("RGBA")
     
     # Remove background
     result = remove(pil_image)
     
-    # Convert result to displayable format
     return result
 
-# Gradio UI
+# Gradio Interface
 title = "AI Background Remover"
 description = "Upload an image and remove its background instantly! Made by Viateur Irasubiza."
 
@@ -28,5 +30,5 @@ iface = gr.Interface(
     allow_flagging="never"
 )
 
-# Launch the app
-iface.launch()
+if __name__ == "__main__":
+    iface.launch()