From fb57fb3cc5aa7299af10ac29037c24deb4d5720f Mon Sep 17 00:00:00 2001
From: spirosmaggioros <spirosmag@ieee.org>
Date: Fri, 10 Jan 2025 23:31:37 +0000
Subject: [PATCH] Fixing setup.py

---
 setup.py | 7 ++++++-
 1 file changed, 6 insertions(+), 1 deletion(-)

diff --git a/setup.py b/setup.py
index 97fb934..73c3e16 100644
--- a/setup.py
+++ b/setup.py
@@ -25,7 +25,12 @@
     url="https://github.com/CBICA/DLMUSE/",
     packages=find_packages(exclude=["tests", ".github"]),
     python_requires=">=3.9",
-    install_requires=required,
+    install_requires=[
+        "torch<=2.3.1",
+        "nnunetv2<=2.5.1",
+        "argparse",
+        "huggingface_hub"
+    ],
     entry_points={"console_scripts": ["DLMUSE = DLMUSE.__main__:main"]},
     classifiers=[
         "Intended Audience :: Developers",