pyproject.toml

[project]
name = "image-classification-model"
version = "0.1.0"
description = "MNIST-compatible image classification model"
requires-python = ">=3.8"
dependencies = [
    "torch>=2.0.0",
    "transformers>=4.30.0",
    "Pillow>=9.0.0",
    "datasets>=2.0.0",
    "accelerate>=0.26.0"
]

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


[tool.setuptools]
packages = ["image_classification_model"]

[tool.setuptools.package-dir]
image_classification_model = "src"

[tool.setuptools.entry_points]
console_scripts = [
    "train = bin.train:main",
    "subtract = bin.subtract:main"
]