| [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" | |
| ] |