Update Dockerfile

Dockerfile

@@ -2,6 +2,7 @@ FROM python:3.9-slim
 
 WORKDIR /app
 
+# Install OS dependencies
 RUN apt-get update && apt-get install -y \
     build-essential \
     curl \
@@ -18,16 +19,22 @@ RUN apt-get update && apt-get install -y \
     zlib1g-dev \
     && rm -rf /var/lib/apt/lists/*
 
+# Copy project files
 COPY requirements.txt ./
 COPY src/ ./src/
 
-RUN pip install numpy && \
+# Install pip packages
+RUN pip install --upgrade pip && \
+    pip install "numpy<2.0" && \
     pip install -r requirements.txt && \
     pip install torchvision==0.15.2 && \
     pip install 'git+https://github.com/facebookresearch/detectron2.git'
 
+# Expose Streamlit port
 EXPOSE 8501
 
-HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health
+# Health check endpoint
+HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health || exit 1
 
+# Start Streamlit app
 ENTRYPOINT ["streamlit", "run", "src/streamlit_app.py", "--server.port=8501", "--server.address=0.0.0.0"]