from setuptools import setup, find_packages

setup(
    name="sketch_nn",
    version="0.1",
    packages=find_packages(),
    install_requires=[
        "opencv-python",
        "numpy",
        "pytesseract",
        "torch",
        "gradio",
        "fastapi",
        "uvicorn",
    ],
    author="Aman Sharma",
    author_email="amananytime07@gmail.com",
    description="A tool to generate PyTorch neural network code from flowchart images",
    long_description=open("README.md").read(),
    long_description_content_type="text/markdown",
    url="https://github.com/yourusername/sketch_nn",
)