Update Dockerfile

Dockerfile

@@ -1,33 +1,30 @@
-# 1. FINAL BASE IMAGE: Use the official, stable PyTorch image
+# 1. Use the stable PyTorch 2.0.1 base image
 FROM pytorch/pytorch:2.0.1-cuda11.7-cudnn8-runtime
 
-# Set working directory inside the container
+# Set working directory
 WORKDIR /app
 
-# Install system dependencies (curl, git, etc.)
+# Install system dependencies
 RUN apt-get update && apt-get install -y \
     curl \
     git \
     build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# ----------------------------------------------------------------------
-# CRITICAL FIX: Install dependencies in the correct order
-# ----------------------------------------------------------------------
-
-# Install all Python packages (including pinned NumPy, which is necessary for spaCy)
+# 2. Copy requirements
 COPY requirements.txt ./
-RUN pip install --no-cache-dir -r requirements.txt
 
-# CRITICAL FIX 2: Download the spaCy language model data
+# 3. CRITICAL: Force reinstall to ensure we get Transformers 4.32.0 and NumPy 1.24.4
+# This overwrites any pre-installed versions in the base image
+RUN pip install --no-cache-dir --upgrade --force-reinstall -r requirements.txt
+
+# 4. Download spaCy model
 RUN python3 -m spacy download en_core_web_sm
 
-# Copy application source code to the container
+# 5. Copy your source code (Ensure your code is in src/streamlit_app.py)
 COPY src/ ./src/
 
-# Expose the port Streamlit runs on
+# 6. Expose port and run
 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"]