Update Dockerfile

Dockerfile

@@ -12,18 +12,14 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # ----------------------------------------------------------------------
-# CRITICAL FIX 1: Pin NumPy AND Force Reinstallation of PyTorch
+# CRITICAL FIX 1: Single, aggressive install step
+# We rely on the base image for PyTorch and force all requirements
+# including the pinned NumPy, to be installed in one go.
 # ----------------------------------------------------------------------
 
-# 1a. Copy requirements and install the pinned NumPy version first
 COPY requirements.txt ./
-RUN pip install --no-cache-dir numpy==1.24.4
-
-# 1b. Force re-install PyTorch to link it correctly with the new NumPy 1.24.4 version
-RUN pip install --no-cache-dir torch
-
-# 1c. Install remaining Python packages from requirements
-RUN pip install --no-cache-dir -r requirements.txt
+# The --upgrade and --force-reinstall flags fix fragmentation errors
+RUN pip install --no-cache-dir --upgrade --force-reinstall -r requirements.txt
 
 # ----------------------------------------------------------------------