From 91f636105564d5464e93dedffd5d4cc99f6b5aaa Mon Sep 17 00:00:00 2001
From: cesnimda <connor.babbington@cesnimda.co.uk>
Date: Sat, 21 Mar 2026 22:12:47 +0100
Subject: [PATCH] remove CUDA

---
 tools/summarizer/Dockerfile       |  9 ++++++++-
 tools/summarizer/requirements.txt | 11 +++++------
 2 files changed, 13 insertions(+), 7 deletions(-)

diff --git a/tools/summarizer/Dockerfile b/tools/summarizer/Dockerfile
index ff1e792..fe1c1e3 100644
--- a/tools/summarizer/Dockerfile
+++ b/tools/summarizer/Dockerfile
@@ -1,7 +1,14 @@
 FROM python:3.11-slim
+ENV PIP_NO_CACHE_DIR=1 \
+    PIP_DISABLE_PIP_VERSION_CHECK=1 \
+    PYTHONUNBUFFERED=1 \
+    TRANSFORMERS_NO_TF=1 \
+    HF_HUB_DISABLE_TELEMETRY=1
 WORKDIR /app
 COPY requirements.txt ./
-RUN pip install --no-cache-dir -r requirements.txt
+RUN pip install --upgrade pip setuptools wheel \
+    && pip install --extra-index-url https://download.pytorch.org/whl/cpu torch==2.5.1 \
+    && pip install -r requirements.txt
 COPY . .
 EXPOSE 8001
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "8001"]
diff --git a/tools/summarizer/requirements.txt b/tools/summarizer/requirements.txt
index 3066c54..be9b26e 100644
--- a/tools/summarizer/requirements.txt
+++ b/tools/summarizer/requirements.txt
@@ -1,6 +1,5 @@
-fastapi>=0.85
-uvicorn[standard]>=0.18
-transformers>=4.30
-torch>=1.13
-cachetools>=5.0
-pydantic>=1.10
+fastapi==0.115.12
+uvicorn[standard]==0.34.0
+transformers==4.48.3
+cachetools==5.5.2
+pydantic==2.10.6