diff --git a/setup.py b/setup.py index baca931..b6c17c3 100644 --- a/setup.py +++ b/setup.py @@ -6,9 +6,14 @@ setup( packages=find_packages(where="src"), package_dir={"": "src"}, install_requires=[ - line.strip() - for line in open("requirements.txt") - if line.strip() and not line.startswith("#") + "torch", + "aiia", + "pillow", + "torchvision", + "scikit-learn", + ], + dependency_links=[ + "git+https://gitea.fabelous.app/Machine-Learning/AIIA.git#egg=aiia" ], python_requires=">=3.10", ) \ No newline at end of file