Update pyproject.toml

backend/pyproject.toml

@@ -4,7 +4,7 @@ version = "0.1.0"
 description = ""
 authors = ["xKhronoz"]
 readme = "README.md"
-packages = [{include = "backend"}]
+packages = [{ include = "backend" }]
 
 [tool.poetry.dependencies]
 python = "^3.11,<3.12"
@@ -15,14 +15,18 @@ pypdf = "^3.17.0"
 python-dotenv = "^1.0.0"
 llama-cpp-python = "^0.2.18"
 transformers = "^4.35.2"
-# For CPU version both Windows and Linux
+# For CPU version: Windows and Linux and MacOS (arm64)
 torch = [
-    { version = "=2.1.1", source = "pypi", platform = "darwin" },
-    { version = "=2.1.1+cpu", source = "pytorch", platform = "linux" },
-    { version = "=2.1.1+cpu", source = "pytorch", platform = "win32" },
+    { url = "https://download.pytorch.org/whl/cpu/torch-2.1.1%2Bcpu-cp311-cp311-win_amd64.whl", markers = "sys_platform == 'win32'" },
+    { url = "https://download.pytorch.org/whl/cpu/torch-2.1.1%2Bcpu-cp311-cp311-linux_x86_64.whl", markers = "sys_platform == 'linux'" },
+    { url = "https://download.pytorch.org/whl/cpu/torch-2.1.1-cp311-none-macosx_11_0_arm64.whl", markers = "sys_platform == 'darwin'" },
 ]
-# torch = { url = "https://download.pytorch.org/whl/cu121/torch-2.1.1%2Bcu121-cp311-cp311-win_amd64.whl"} # For Windows with Cuda 12.1
-# torch = { url = "https://download.pytorch.org/whl/cu121/torch-2.1.1%2Bcu121-cp311-cp311-linux_x86_64.whl"} # For Linux with Cuda 12.1
+## For GPU version: Windows and Linux and MacOS (arm64)
+# torch = [
+#     { url = "https://download.pytorch.org/whl/cu121/torch-2.1.1%2Bcu121-cp311-cp311-win_amd64.whl", markers = "sys_platform == 'win32'" },
+#     { url = "https://download.pytorch.org/whl/cu121/torch-2.1.1%2Bcu121-cp311-cp311-linux_x86_64.whl", markers = "sys_platform == 'linux'" },
+#     { url = "https://download.pytorch.org/whl/cu121/torch-2.1.1-cp311-none-macosx_11_0_arm64.whl", markers = "sys_platform == 'darwin'" },
+# ]
 docx2txt = "^0.8"