from setuptools import setup, find_packages

setup(
    name="windblade_seg",
    version="0.1.0",
    description="Wind turbine blade segmentation inference",
    packages=find_packages(),
    include_package_data=True,
    python_requires=">=3.8",
    install_requires=[
        "torch",
        "torchvision",
        "opencv-python",
        "Pillow",
        "segmentation-models-pytorch==0.3.3",
    ],
)