Update app.py

app.py

@@ -4,8 +4,13 @@ from PIL import Image
 import io
 
 def remove_bg(input_image):
-    output = remove(input_image)
-    return Image.open(io.BytesIO(output))
+    try:
+        if input_image is None:
+            return "Error: No image received"
+        output = remove(input_image)
+        return Image.open(io.BytesIO(output))
+    except Exception as e:
+        return f"Error: {str(e)}"
 
 demo = gr.Interface(fn=remove_bg, inputs=gr.Image(type="pil"), outputs="image")
 demo.launch()