File size: 1,618 Bytes
ffaa4dd |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 |
[build-system]
requires = ["setuptools", "wheel"]
build-backend = "setuptools.build_meta"
[project]
name = "modelapi"
version = "1.0.0"
description = "This project implements a container for a fine-tuned audio enhancement model."
readme = { file = "README.md", content-type = "text/markdown" }
license = { file = "LICENSE" }
classifiers = [
"Development Status :: 3 - Beta",
"Intended Audience :: Developers",
"Topic :: Software Development :: Build Tools",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3 :: Only",
"Programming Language :: Python :: 3.10",
"Topic :: Scientific/Engineering",
"Topic :: Scientific/Engineering :: Mathematics",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"Topic :: Software Development",
"Topic :: Software Development :: Libraries",
"Topic :: Software Development :: Libraries :: Python Modules"
]
requires-python = ">=3.10,<3.12"
dependencies = [
"fastapi==0.115.5",
"uvicorn==0.32.0",
"python-multipart==0.0.17",
# Added packages
"gdown",
"h5py",
"ipympl",
"librosa",
"ninja",
"numpy<2.0",
"pandas",
"pesq",
"pillow",
"protobuf",
"pyarrow",
"pyroomacoustics",
"pystoi",
"pytorch-lightning",
"scipy",
"sdeint",
"setuptools", # Already included in build-system but harmless here
"seaborn",
"torch",
"torch-ema",
"torchaudio",
"torchinfo",
"torchvision",
"torchsde",
"tqdm",
"wandb",
"transformers"
]
[tool.setuptools.packages.find]
include = ["app", "model"]
|