Update Dockerfile

Dockerfile

@@ -1,45 +1,34 @@
+# Stage 1 - Builder
 FROM python:3.10-slim AS builder
-
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install build tools
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential cmake git curl \
     && rm -rf /var/lib/apt/lists/*
 
 WORKDIR /app
-
-# Copy and install Python dependencies
 COPY requirements.txt .
 RUN python -m venv /opt/venv && \
     /opt/venv/bin/pip install --upgrade pip && \
     /opt/venv/bin/pip install --no-cache-dir -r requirements.txt
 
-# Pre-download YOLOv8 model weights into writable path
-RUN mkdir -p /app/models && \
-    curl -L -o /app/models/yolov8n.pt \
+# Pre-download YOLOv8 weights to the exact path Ultralytics expects
+RUN mkdir -p /root/.cache/torch/hub/checkpoints && \
+    curl -L -o /root/.cache/torch/hub/checkpoints/yolov8n.pt \
     https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt
 
-# -------- RUNTIME IMAGE --------
+# Stage 2 - Final
 FROM python:3.10-slim
-
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PATH="/opt/venv/bin:$PATH"
-ENV EASYOCR_MODULE_PATH=/app/.EasyOCR
-ENV EASYOCR_CACHE_DIR=/app/.EasyOCR
-ENV YOLO_CONFIG_DIR=/app/.yolo
 
-# Install minimal OpenCV runtime dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
-# Copy venv and preloaded model
 COPY --from=builder /opt/venv /opt/venv
-COPY --from=builder /app/models/yolov8n.pt /app/models/yolov8n.pt
-
-# Create writable dirs for EasyOCR & YOLO configs
-RUN mkdir -p /app/.EasyOCR /app/.yolo && chmod -R 777 /app/.EasyOCR /app/.yolo
+COPY --from=builder /root/.cache/torch/hub/checkpoints/yolov8n.pt \
+     /root/.cache/torch/hub/checkpoints/yolov8n.pt
 
 WORKDIR /app
 COPY . .