Update app.py

app.py

@@ -15,11 +15,16 @@ def image_to_data_url(img: Image.Image) -> str:
     img.save(buffered, format="PNG")
     img_str = base64.b64encode(buffered.getvalue()).decode("utf-8")
     return f"data:image/png;base64,{img_str}"
+def base64_to_image(data_url):
+    header, encoded = data_url.split(",", 1)
+    binary_data = base64.b64decode(encoded)
+    image = Image.open(BytesIO(binary_data))
+    return image
 
 @app.route("/remove_background", methods=["POST"])
 def remove_background():
     data = request.json
-    image_url = data.get("image_url")
+    image_url = base64_to_image(data.get("image_url"))
 
     if not image_url:
         return jsonify({"error": "image_url is required"}), 400