aipluto-backend / pyproject.toml
andreskoenig's picture
deploy from cli
d32533a verified
Raw
History Blame Contribute Delete
1.05 kB
[build-system]
requires = ["setuptools>=68", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "find-my-dog-backend"
version = "0.1.0"
description = "Backend for Find My Dog — detection + DINOv2 embedding + pgvector search."
requires-python = ">=3.11"
dependencies = [
# web
"fastapi>=0.110",
"uvicorn[standard]>=0.27",
"python-multipart>=0.0.9",
"pydantic>=2.6",
"pydantic-settings>=2.2",
# db
"SQLAlchemy>=2.0",
"psycopg[binary]>=3.1",
"alembic>=1.13",
"pgvector>=0.2.5",
# ml — torch CPU wheels come from --extra-index-url at install time
"torch==2.4.1",
"torchvision==0.19.1",
"transformers>=4.40,<4.46",
"ultralytics>=8.1",
# imaging
"pillow>=10.2",
"numpy>=1.26,<2.0",
"opencv-python-headless>=4.9",
# storage
"minio>=7.2",
# seeding
"datasets>=2.18",
"huggingface_hub>=0.22",
"tqdm>=4.66",
# tests
"pytest>=8.0",
"httpx>=0.27",
]
[tool.setuptools.packages.find]
where = ["."]
include = ["app*", "scripts*"]