Update Dockerfile

Dockerfile

@@ -1,21 +1,31 @@
 FROM python:3.12-slim
 
+ENV PIP_NO_CACHE_DIR=1 \
+    PYTHONDONTWRITEBYTECODE=1 \
+    PYTHONUNBUFFERED=1
 
 WORKDIR /app
 
-RUN apt-get update && apt-get install -y \
+# Minimal system deps; no recommends = smaller image
+RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     curl \
     git \
     && rm -rf /var/lib/apt/lists/*
 
+# Install Python deps first to leverage Docker layer caching
 COPY requirements.txt ./
-COPY src/ ./src/
+RUN pip install --upgrade pip setuptools wheel \
+ && pip install --prefer-binary -r requirements.txt
 
-RUN pip3 install -r requirements.txt
+# Copy application code and the model artifact
+COPY src/ ./src/
+COPY classifier.pkl ./  # <-- ensures joblib.load("classifier.pkl") works
 
 EXPOSE 8501
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Healthcheck (retry-friendly)
+HEALTHCHECK --interval=30s --timeout=3s --retries=3 \
+  CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
-ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]
+ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]