final modifications for Dockerfile and requirements.txt

Dockerfile

@@ -24,6 +24,9 @@ ENV PYTHONUNBUFFERED=1 \
 # Upgrade pip & install wheel
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel
 
+# Install CPU-only PyTorch
+RUN pip install --no-cache-dir torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cpu
+
 # Install project dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 

requirements.txt

@@ -1,17 +1,8 @@
-# Extra index for PyTorch CPU packages
---extra-index-url https://download.pytorch.org/whl/cpu
-
 # Core LLM + Transformers
-numpy>=1.26.0
 transformers[sentencepiece]>=4.40.0
 sentence-transformers>=2.6.0
 accelerate>=0.30.0
 
-# ML stack
-torch==2.2.2+cpu
-torchvision==0.17.2+cpu
-torchaudio==2.2.2+cpu
-
 # LangChain + LangGraph
 langchain>=0.2.1
 langgraph>=0.0.45