from setuptools import setup, find_packages

setup(
    name="neuroprompt",
    version="0.1.1",
    packages=find_packages(),
    install_requires=[
        "torch>=2.0.0",
        "openai>=1.0.0",
        "llmlingua>=0.1.0",
        "tiktoken>=0.5.0",
        "nltk>=3.8.0",
        "numpy>=1.24.0",
        "rouge-score>=0.1.2",
    ],
    python_requires=">=3.8",
)