model-loader / pyproject.toml
sarpreetsingh3131's picture
model-loader v0.1.0
bbdeb86
Raw History Blame Contribute Delete
955 Bytes
[build-system]
requires = ["setuptools>=68"]
build-backend = "setuptools.build_meta"
[project]
name = "aipanjab-loader"
version = "0.1.0"
description = "AI Panjab models in one line: load(...) any aipanjab model from Hugging Face and run it on your machine"
readme = "README.md"
requires-python = ">=3.10"
license = "CC-BY-4.0"
authors = [{ name = "AI Panjab" }]
# what the models' own code needs to run (each model's inference.py): every aipanjab model runs on these
dependencies = [
"huggingface_hub>=0.25",
"numpy<2.3",
"onnxruntime>=1.17",
"pillow",
"pyyaml",
]
[project.optional-dependencies]
# training: pip install "aipanjab-loader[train] @ git+https://huggingface.co/aipanjab/model-loader"
train = ["torch>=2.4", "keras>=3.8", "keras-hub>=0.19", "regex", "onnx", "onnxscript", "datasets"]
[project.urls]
Homepage = "https://aipanjab.com"
Models = "https://huggingface.co/aipanjab"
[tool.setuptools]
packages = ["aipanjab"]