from setuptools import setup, find_packages

setup(
    name="sdg-predict",
    version="0.2",
    packages=find_packages(),
    install_requires=[
        "transformers>=4.36",
        "torch>=2.0",
        "tqdm",
    ],
    entry_points={
        "console_scripts": [
            "sdg-predict = sdg_predict.cli_predict:main"
        ]
    },
    author="Simon Clematide",
    description="Command-line prediction for SDG SciBERT classifier",
    python_requires=">=3.8",
)