setup.py

from setuptools import setup, find_packages

setup(
    name="My_ai",
    version="0.1",
    packages=find_packages(),
    install_requires=[
        'torch',
        'transformers',
        'flask',
        'python-dotenv',
        'tensorflow',  # Added TensorFlow
        # Add other dependencies here
    ],
)