pyproject.toml

[project]
name = "modelzilla"
version = "0.1.0"
description = "It is a lightweight Python package that enables developers to transform any AI model into a fully functional Command-Line Interface (CLI) plugin"
authors = [{name = "David Redo", email = "dredonieto@gmail.com"}]
readme = "README.md"
license = {file = "LICENSE"}
classifiers = [
    "License :: OSI Approved :: Apache Software License",
    "Programming Language :: Python",
    "Programming Language :: Python :: 3",
    "Programming Language :: Python :: 3.10",
    "Programming Language :: Python :: 3.11",
    "Programming Language :: Python :: 3.12",
]
dependencies = [
    "supervision>=0.25.1",
]
requires-python = ">=3.10,<3.13"

[project.optional-dependencies]
hf = [
    "transformers>=4.47.1",
    "torch>=2.1.0",
    "timm>=1.0.13",
    "numpy<2.0", # Curretly some problems arises with numpy>2 and transformers
]

[project.scripts]
modelzilla = "modelzilla.main:main"

[project.urls]
Homepage = "https://github.com/David-rn/modelzilla/"
Issues = "https://github.com/David-rn/modelzilla/issues"

[tool.poetry]
packages = [{include = "modelzilla"}]

[tool.poetry.group.dev.dependencies]
ruff = "^0.8.5"
pytest = "^8.3.4"
tox = "^4.23.2"

[tool.poetry.group.build.dependencies]
twine = "^6.0.1"
wheel = "^0.45.1"
build = "^1.2.2.post1"

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