File size: 476 Bytes
9d36a4d fed6436 9d36a4d |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 |
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",
)
|