from setuptools import setup, find_packages

setup(
    name="qwen_classifier",
    version="0.1",
    packages=find_packages(),
    package_data={
        'qwen_classifier': ['test_jsons.zip'],
    },
    include_package_data=True,
    install_requires=[
        'torch',
        'transformers',
        'click',
        'scikit-learn',
        'huggingface_hub',
        'requests',
        'pandas',
        'pydantic',
        'datasets',
        'scikit-learn',
        'matplotlib'
    ],
    entry_points={
        'console_scripts': [
            'qwen-clf=qwen_classifier.cli:cli',
        ],
    },
)