smaller dockerfile

Author: NoCapCbas

dockerfile.prod

@@ -1,37 +1,47 @@
-# Use an official Python runtime as a parent image
-FROM python:3.12-slim
+# Build stage
+FROM python:3.12-slim-bullseye as builder
 
-# Set working directory in the container
 WORKDIR /app
 
-# Install system dependencies including Ollama
+# Install build dependencies
 RUN apt-get update && apt-get install -y \
     curl \
+    build-essential \
     && rm -rf /var/lib/apt/lists/*
 
-# Install Ollama
-RUN curl https://ollama.ai/install.sh | sh
-
-# Copy the requirements
+# Copy and install requirements with CPU-only PyTorch
 COPY requirements.txt .
+RUN pip install --no-cache-dir -r requirements.txt \
+    && pip uninstall -y torch \
+    && pip install --no-cache-dir torch --index-url https://download.pytorch.org/whl/cpu
+
+# Final stage
+FROM python:3.12-slim-bullseye
+
+WORKDIR /app
+
+# Install only essential runtime dependencies
+RUN apt-get update && apt-get install -y \
+    curl \
+    && rm -rf /var/lib/apt/lists/* \
+    && curl https://ollama.ai/install.sh | sh
 
-# Install Python dependencies
-RUN pip install --no-cache-dir -r requirements.txt
+# Copy only necessary Python packages
+COPY --from=builder /usr/local/lib/python3.12/site-packages/ /usr/local/lib/python3.12/site-packages/
 
-# Copy the application
-COPY . .
+# Copy application files
+COPY app.py .
+COPY dogs_cleaned.csv .
 
-# Start Ollama service and pull the model
+# Start Ollama, pull model, and clean up unnecessary files
 RUN ollama serve & \
     sleep 10 && \
-    ollama pull mistral
+    ollama pull mistral && \
+    rm -rf /root/.cache
 
-# Expose the port Streamlit runs on
 EXPOSE 8080
 
-# Create a shell script to start both Ollama and Streamlit
-RUN echo '#!/bin/bash\nollama serve & sleep 5 && streamlit run app.py --server.address=0.0.0.0 --server.port=8080' > start.sh
-RUN chmod +x start.sh
+RUN echo '#!/bin/bash\nollama serve & sleep 5 && streamlit run app.py --server.address=0.0.0.0 --server.port=8080' > start.sh \
+    && chmod +x start.sh
 
-# Command to run the script
 CMD ["./start.sh"]