setup.py

@@ -6,9 +6,14 @@ setup(
     packages=find_packages(where="src"),
     package_dir={"": "src"},
     install_requires=[
-        line.strip()
+        "torch",
-        for line in open("requirements.txt")
+        "aiia",
-        if line.strip() and not line.startswith("#")
+        "pillow",
+        "torchvision",
+        "scikit-learn",
+    ],
+    dependency_links=[
+        "git+https://gitea.fabelous.app/Machine-Learning/AIIA.git#egg=aiia"
     ],
     python_requires=">=3.10",
 )