| 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", | |
| ) | |