Update app.py

app.py

@@ -7,6 +7,9 @@ import io
 # Function to remove background
 # -------------------------------
 def remove_bg(image: Image.Image):
+    if image is None:
+        return None
+
     # Convert PIL image to bytes
     img_byte_arr = io.BytesIO()
     image.save(img_byte_arr, format='PNG')
@@ -25,10 +28,11 @@ def remove_bg(image: Image.Image):
 # -------------------------------
 demo = gr.Interface(
     fn=remove_bg,
-    inputs=gr.Image(type="pil", label="Upload an image"),
+    inputs=gr.Image(type="pil", label="Upload any image"),
     outputs=gr.Image(type="pil", label="Background Removed"),
-    title="Background Remover",
-    description="Upload an image to remove its background using the rembg library.",
+    title="Universal Background Remover",
+    description="Upload any image and the background will be automatically removed using rembg.",
+    allow_flagging="never",
 )
 
 # -------------------------------