from setuptools import setup, find_packages

setup(
    name='tasks',
    version='0.1',
    packages=find_packages(),
    install_requires=[
        'fastapi>=0.68.0',
        'uvicorn>=0.15.0',
        'codecarbon>=2.3.1',
        'datasets>=2.14.0',
        'scikit-learn>=1.0.2',
        'pydantic>=1.10.0',
        'python-dotenv>=1.0.0',
        'gradio>=4.0.0',
        'requests>=2.31.0',
        'librosa==0.10.2.post1',
        'numpy==1.26.4',
        'ultralytics==8.3.68',
        'ultralytics-thop==2.0.14',
        #opencv-python==4.11.0.86
        'python-dotenv==1.0.0',
        'matplotlib==3.8.1',
        'onnx==1.17.0',
        'tensorrt==10.8.0.43',
        'tensorrt_cu12==10.8.0.43',
        'tensorrt_cu12_bindings==10.8.0.43',
        'tensorrt_cu12_libs==10.8.0.43',
        'optimum[onnxruntime-gpu]==1.24.0'
    ]
)