Update Dockerfile

Dockerfile

@@ -12,31 +12,36 @@ 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 weights to the exact path Ultralytics expects
+# ✅ Pre-download YOLOv8n weights to Ultralytics expected cache path
 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
 
-# Stage 2 - Final
+# Stage 2 - Final Image
 FROM python:3.10-slim
 ENV DEBIAN_FRONTEND=noninteractive
 ENV PATH="/opt/venv/bin:$PATH"
 
-# Force EasyOCR to store in /app/.EasyOCR instead of /
+# ✅ Fix EasyOCR cache location
 ENV EASYOCR_MODULE_PATH=/app/.EasyOCR
 ENV EASYOCR_CACHE_DIR=/app/.EasyOCR
 
+# ✅ Force YOLO to use its cache folder, not cwd
+ENV YOLO_CONFIG_DIR=/app/.yolo
+
 RUN apt-get update && apt-get install -y --no-install-recommends \
     libgl1 libglib2.0-0 \
     && rm -rf /var/lib/apt/lists/*
 
 COPY --from=builder /opt/venv /opt/venv
-COPY --from=builder /root/.cache/torch/hub/checkpoints/yolov8n.pt \
-     /root/.cache/torch/hub/checkpoints/yolov8n.pt
+COPY --from=builder /root/.cache /root/.cache
 
 WORKDIR /app
-# Create EasyOCR cache folder and set permissions
-RUN mkdir -p /app/.EasyOCR && chmod -R 777 /app/.EasyOCR
+
+# ✅ Create required cache folders and fix permissions
+RUN mkdir -p /app/.EasyOCR /app/.yolo && \
+    chmod -R 777 /app && \
+    chmod -R 777 /root/.cache
 
 COPY . .