[build-system]
requires = ["setuptools>=77", "wheel"]
build-backend = "setuptools.build_meta"

[project]
name = "ax-doclayout"
version = "1.2.0.post3"
description = "AX DocLayout page-raster layout inference."
readme = "README.md"
requires-python = ">=3.12,<3.14"
license = "AGPL-3.0-only"
license-files = ["LICENSE", "NOTICE"]
authors = [{ name = "AutomatosX" }]
dependencies = ["huggingface-hub>=0.36,<2", "numpy>=1.26,<3", "pillow>=12.2,<13"]

[project.optional-dependencies]
onnx = [
  "onnxruntime-gpu[cuda,cudnn]==1.23.2",
  "nvidia-cuda-runtime-cu12==12.8.90",
  "nvidia-cuda-nvrtc-cu12==12.8.93",
  "nvidia-cufft-cu12==11.3.3.83",
  "nvidia-curand-cu12==10.3.9.90",
  "nvidia-cublas-cu12==12.8.4.1",
  "nvidia-cudnn-cu12==9.10.2.21",
]
torch = ["torch>=2.12,<2.13", "safetensors>=0.4,<1"]

[project.scripts]
ax-doclayout = "ax_doclayout.cli:main"

[tool.setuptools.packages.find]
where = ["src"]

[tool.setuptools.package-data]
ax_doclayout = ["py.typed"]