Update Dockerfile

Dockerfile

@@ -1,29 +1,28 @@
-# 1. Use a base image with PyTorch and CUDA pre-installed.
-# This eliminates the long 'pip install torch' and compilation time.
+
 FROM nvcr.io/nvidia/pytorch:23.09-py3
 
-# Set working directory inside the container
+
 WORKDIR /app
 
-# Install system dependencies (like curl and git)
+
 RUN apt-get update && apt-get install -y \
     curl \
     git \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy requirements file and install remaining packages
+
 COPY requirements.txt ./
 RUN pip install --no-cache-dir -r requirements.txt
 
-# CRITICAL FIX: Download the spaCy language model
+
 RUN python3 -m spacy download en_core_web_sm
 
-# Copy 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"]