api/src/constants.py

ALLOWED_EXTENSIONS = {".jpg", ".jpeg", ".png", ".webp"}

SUPPORTED_MODELS = [
    # "u2net",
    # "u2netp",
    # "silueta",
    # "isnet-general-use",
    # "sam",
    "birefnet-general",
    "birefnet-general-lite",
    "bria-rmbg",
]

MAX_WORKERS = 4