Update Dockerfile

Dockerfile

@@ -1,10 +1,8 @@
-# Base image
 FROM python:3.10-slim
 
-# Prevent interactive prompts during installs
 ENV DEBIAN_FRONTEND=noninteractive
 
-# Install system dependencies (minimal for EasyOCR + Pillow)
+# Install system dependencies
 RUN apt-get update && apt-get install -y --no-install-recommends \
     build-essential \
     cmake \
@@ -12,22 +10,22 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     python3-opencv \
     tesseract-ocr \
     libtesseract-dev \
+    wget \
     && rm -rf /var/lib/apt/lists/*
 
-# Set work directory
 WORKDIR /app
 
-# Copy requirements first (to leverage caching)
+# Copy requirements first for caching
 COPY requirements.txt .
 
-# Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# Copy app files
+# Pre-download YOLO weights into /app
+RUN wget https://github.com/ultralytics/assets/releases/download/v8.3.0/yolov8n.pt -O /app/yolov8n.pt
+
+# Copy application code
 COPY app.py .
 
-# Expose port for HF Spaces
 EXPOSE 7860
 
-# Run app
 CMD ["python", "app.py"]