from flask import Flask, request, Response
from rembg import remove
from rembg.session_factory import new_session
import io
app = Flask(__name__)
# FULL U²-Net
session = new_session("u2net")
HTML = """
Xlnk RMBG - AI Background Remover
Drop image from anywhere
"""
@app.route("/", methods=["GET", "POST"])
def index():
if request.method == "POST":
file = request.files.get("file")
if not file:
return "No file", 400
input_bytes = file.read()
output_bytes = remove(input_bytes, session=session, alpha_matting=False)
return Response(
output_bytes,
mimetype="image/png",
headers={"Content-Disposition": "attachment; filename=xlnk-rmbg.png"}
)
return HTML
if __name__ == "__main__":
app.run(host="0.0.0.0", port=7860)