Update Dockerfile

Dockerfile

@@ -1,5 +1,5 @@
-# 1. Use an official PyTorch base image
-FROM nvcr.io/nvidia/pytorch:23.09-py3
+# 1. FINAL BASE IMAGE: Use the official, stable PyTorch image
+FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
 # Set working directory inside the container
 WORKDIR /app
@@ -12,16 +12,12 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/*
 
 # ----------------------------------------------------------------------
-# 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.
+# CRITICAL FIX: Install dependencies in the correct order
 # ----------------------------------------------------------------------
 
+# Install all Python packages (including pinned NumPy, which is necessary for spaCy)
 COPY requirements.txt ./
-# The --upgrade and --force-reinstall flags fix fragmentation errors
-RUN pip install --no-cache-dir --upgrade --force-reinstall -r requirements.txt
-
-# ----------------------------------------------------------------------
+RUN pip install --no-cache-dir -r requirements.txt
 
 # CRITICAL FIX 2: Download the spaCy language model data
 RUN python3 -m spacy download en_core_web_sm