Upgrade dependencies

Dockerfile

@@ -14,7 +14,7 @@ ENV POETRY_VENV=/app/.venv
 
 RUN python3 -m venv $POETRY_VENV \
     && $POETRY_VENV/bin/pip install -U pip setuptools \
-    && $POETRY_VENV/bin/pip install poetry==1.6.1
+    && $POETRY_VENV/bin/pip install poetry==2.1.1
 
 ENV PATH="${PATH}:${POETRY_VENV}/bin"
 

Dockerfile.gpu

@@ -28,7 +28,7 @@ RUN ln -s -f /usr/bin/python${PYTHON_VERSION} /usr/bin/python3 && \
 
 RUN python3 -m venv $POETRY_VENV \
     && $POETRY_VENV/bin/pip install -U pip setuptools \
-    && $POETRY_VENV/bin/pip install poetry==1.6.1
+    && $POETRY_VENV/bin/pip install poetry==2.1.1
 
 ENV PATH="${PATH}:${POETRY_VENV}/bin"
 
@@ -45,7 +45,7 @@ COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui.css swagger-ui-assets/sw
 COPY --from=swagger-ui /usr/share/nginx/html/swagger-ui-bundle.js swagger-ui-assets/swagger-ui-bundle.js
 
 RUN poetry install
-RUN $POETRY_VENV/bin/pip install torch==1.13.1+cu117 -f https://download.pytorch.org/whl/torch
+RUN $POETRY_VENV/bin/pip install torch==2.6.0 --index-url https://download.pytorch.org/whl/cu121
 
 RUN $POETRY_VENV/bin/pip install pandas transformers nltk pyannote.audio
 RUN git clone --depth 1 https://github.com/m-bain/whisperX.git \

pyproject.toml

@@ -11,54 +11,64 @@ packages = [{ include = "app" }]
 [tool.poetry.scripts]
 whisper-asr-webservice = "app.webservice:start"
 
+# Package Sources
 [[tool.poetry.source]]
 name = "pytorch"
 url = "https://download.pytorch.org/whl/cpu"
 priority = "explicit"
 
+# Core Dependencies
 [tool.poetry.dependencies]
 python = "^3.10"
+fastapi = "^0.115.8"
 uvicorn = { extras = ["standard"], version = "^0.34.0" }
-tqdm = "^4.67.1"
 python-multipart = "^0.0.20"
 ffmpeg-python = "^0.2.0"
-fastapi = "^0.115.6"
-llvmlite = "^0.43.0"
-numba = "^0.60.0"
+
+# ML Dependencies
 openai-whisper = "^20240930"
 faster-whisper = "^1.1.1"
+tqdm = "^4.67.1"
+llvmlite = "^0.44.0"
+numba = "^0.61.0"
+
+# PyTorch Dependencies (Platform Specific)
 torch = [
-  { markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1-cp310-none-macosx_11_0_arm64.whl" },
-  { markers = "sys_platform == 'linux' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1-cp310-none-macosx_11_0_arm64.whl" },
-  { markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1-cp310-none-macosx_10_9_x86_64.whl" },
-  { markers = "sys_platform == 'linux' and platform_machine == 'aarch64'", url = "https://download.pytorch.org/whl/torch-1.13.1-cp310-cp310-manylinux2014_aarch64.whl" },
-  { markers = "sys_platform == 'linux' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1%2Bcpu-cp310-cp310-linux_x86_64.whl" },
-  { markers = "sys_platform == 'win' and platform_machine == 'amd64'", url = "https://download.pytorch.org/whl/cpu/torch-1.13.1%2Bcpu-cp310-cp310-win_amd64.whl" },
+  { markers = "sys_platform == 'darwin' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" },
+  { markers = "sys_platform == 'linux' and platform_machine == 'arm64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" },
+  { markers = "sys_platform == 'darwin' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0-cp310-none-macosx_11_0_arm64.whl" },
+  { markers = "sys_platform == 'linux' and platform_machine == 'aarch64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-manylinux_2_28_aarch64.whl" },
+  { markers = "sys_platform == 'linux' and platform_machine == 'x86_64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-linux_x86_64.whl" },
+  { markers = "sys_platform == 'win' and platform_machine == 'amd64'", url = "https://download.pytorch.org/whl/cpu/torch-2.6.0%2Bcpu-cp310-cp310-win_amd64.whl" },
 ]
 
+
+# Development Dependencies
 [poetry.group.dev.dependencies]
 pytest = "^8.3.4"
-ruff = "^0.8.3"
-black = "^24.10.0"
+ruff = "^0.9.6"
+black = "^25.1.0"
 mkdocs = "^1.6.1"
-mkdocs-material = "^9.5.49"
-pymdown-extensions = "^10.12"
+mkdocs-material = "^9.6.4"
+pymdown-extensions = "^10.14.3"
 
+# Build System Configuration
 [build-system]
 requires = ["poetry-core>=1.0.0"]
 build-backend = "poetry.core.masonry.api"
 
+# Code Formatting Configuration
 [tool.black]
 skip-string-normalization = true
 line-length = 120
 
+# Linting Configuration
 [tool.ruff]
-# Same as Black.
 line-length = 120
 
 [tool.ruff.lint]
 select = [
-  "E", # pycodestyle errors (settings from FastAPI, thanks, @tiangolo!)
+    "E",  # pycodestyle errors
     "W",  # pycodestyle warnings
     "F",  # pyflakes
     "I",  # isort