[project]
name = "rmbg-as-a-service"
version = "0.1.0"
description = "Background removal as a GPU-accelerated API"
readme = "README.md"
requires-python = ">=3.12,<3.13"
dependencies = [
    "torch==2.5.1",
    "torchvision==0.20.1",
    "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]
birefnet-service = "birefnet_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/birefnet_service"]

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

[tool.uv.sources]
torch = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]
torchvision = [{ index = "pytorch-cu124", marker = "sys_platform == 'linux'" }]

[tool.ruff]
line-length = 100