[project]
name = "rmbg-as-a-service"
version = "0.0.3"
description = "Background removal as a GPU-accelerated API"
readme = "README.md"
requires-python = ">=3.12"
dependencies = [
    "torch==2.11.0",
    "torchvision==0.26.0",
    "transformers>=4.44,<5",
    "timm>=1.0.0",
    "einops>=0.8.0",
    "kornia>=0.7.0",
    "pillow>=10.0.0",
    "numpy>=1.26",
    "litserve>=0.2.4",
]

[project.scripts]
rmbg-as-a-service = "rmbg_as_a_service.server:run"

[dependency-groups]
dev = ["ruff>=0.6.0"]

[build-system]
requires = ["hatchling"]
build-backend = "hatchling.build"

[tool.hatch.build.targets.wheel]
packages = ["src/rmbg_as_a_service"]

[tool.hatch.build.targets.sdist]
include = [
    "src/rmbg_as_a_service",
    "README.md",
    "pyproject.toml",
]

# BiRefNet (torch) needs CUDA wheels; pull torch/torchvision from the PyTorch index.
[[tool.uv.index]]
name = "pytorch-cu126"
url = "https://download.pytorch.org/whl/cu126"
explicit = true

[tool.uv.sources]
torch = { index = "pytorch-cu126" }
torchvision = { index = "pytorch-cu126" }

[tool.ruff]
line-length = 100