pyproject.toml

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