[tool.poetry]
name = "detect-it"
version = "0.1.0"
description = "Detect all the things"
authors = ["slashtechno <77907286+slashtechno@users.noreply.github.com>"]
license = "MIT"
readme = "README.md"
packages = [{include = "detect_it"}]

[tool.poetry.dependencies]
python = "^3.10"
python-dotenv = "^1.0.0"
httpx = "^0.25.0"
opencv-python = "^4.8.1.78"
ultralytics = "^8.0.190"
hjson = "^3.1.0"
numpy = "^1.23.2"
torch = [
  { version = "^2.0.0+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
  { version = "^2.0.0+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
]
torchaudio = [
  { version = "^2.0.0+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
  { version = "^2.0.0+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
]
torchvision = [
  { version = "^0.15+cu118", source = "torch_cu118", markers = "extra=='cuda'" },
  { version = "^0.15+cpu", source = "torch_cpu", markers = "extra!='cuda'" },
]


[tool.poetry.group.dev.dependencies]
black = "^23.9.1"
ruff = "^0.0.291"
ipykernel = "^6.25.2"


[[tool.poetry.source]]
name = "torch_cpu"
url = "https://download.pytorch.org/whl/cpu"
priority = "supplemental"

[[tool.poetry.source]]
name = "torch_cu118"
url = "https://download.pytorch.org/whl/cu118"
priority = "supplemental"

[tool.poetry.extras]
cuda = []

[[tool.poetry.source]]
name = "PyPI"
priority = "primary"

[build-system]
requires = ["poetry-core"]
build-backend = "poetry.core.masonry.api"