setup.py

import os
from setuptools import setup, find_packages

# Read requirements from requirements.txt if present
install_requires = []
req_path = "requirements.txt"
if os.path.exists(req_path):
    with open(req_path, "r") as f:
        install_requires = [line.strip() for line in f.readlines() if line.strip()]
else:
    install_requires = [
        "torch>=2.0.0",
        "torchvision>=0.15.0",
        "transformers>=4.38.0",
        "datasets>=2.18.0",
        "tqdm>=4.66.0",
        "numpy>=1.24.0",
        "matplotlib>=3.7.0",
        "opencv-python>=4.8.0",
        "Pillow>=10.0.0",
        "scipy>=1.10.0",
        "albumentations>=1.3.0",
        "pycocotools>=2.0.6",
    ]

setup(
    name="prompt_guided_segmentation",
    version="0.1.0",
    description="Prompt-Guided Image Segmentation Lightweight Architecture",
    author="halim-cv",
    packages=find_packages(),
    install_requires=install_requires,
    python_requires=">=3.8",
)