Update Dockerfile

Dockerfile

@@ -1,23 +1,41 @@
-
+# 1. Use an official PyTorch base image
 FROM nvcr.io/nvidia/pytorch:23.09-py3
 
+# Set working directory inside the container
 WORKDIR /app
 
+# Install system dependencies (curl, git, etc.)
 RUN apt-get update && apt-get install -y \
     curl \
     git \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
+# ----------------------------------------------------------------------
+# CRITICAL FIX 1: Pin NumPy AND Force Reinstallation of PyTorch
+# ----------------------------------------------------------------------
+
+# 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
 
+# ----------------------------------------------------------------------
+
+# CRITICAL FIX 2: Download the spaCy language model data
 RUN python3 -m spacy download en_core_web_sm
 
+# Copy application source code to the container
 COPY src/ ./src/
 
+# Expose the port Streamlit runs on
 EXPOSE 8501
 HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
 
+# Command to run the Streamlit application
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]