Update Dockerfile

Dockerfile

@@ -11,29 +11,41 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libxrender1 \
     && rm -rf /var/lib/apt/lists/*
 
-# 2. Set the working directory
+# 2. Set working directory
 WORKDIR /app
 
-# 3. Copy requirements and install them
+# 3. Copy requirements
 COPY requirements.txt .
+
+# 4. Install Python dependencies
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 4. Pre-download the Fashion-CLIP model during the build
+# 5. Download Fashion-CLIP during build
 RUN python -c "from transformers import CLIPModel, CLIPProcessor; \
-    model = CLIPModel.from_pretrained('patrickjohncyh/fashion-clip'); \
-    processor = CLIPProcessor.from_pretrained('patrickjohncyh/fashion-clip'); \
+    CLIPModel.from_pretrained('patrickjohncyh/fashion-clip'); \
+    CLIPProcessor.from_pretrained('patrickjohncyh/fashion-clip'); \
     print('Fashion-CLIP model preloaded!')"
 
-# 5. Copy application code
+# 6. Download YOLO model during build
+RUN python -c "from ultralytics import YOLO; \
+    YOLO('yolov8n.pt'); \
+    print('YOLO model preloaded!')"
+
+# 7. Copy application code
 COPY . .
 
-# 6. Create a non-root user
+# 8. Create non-root user
 RUN useradd -m -u 1000 user
+
+# 9. Give the application user access to the application directory
+RUN chown -R user:user /app
+
 USER user
 
 ENV HOME=/home/user \
     PATH=/home/user/.local/bin:$PATH \
-    PYTHONUNBUFFERED=1
+    PYTHONUNBUFFERED=1 \
+    YOLO_CONFIG_DIR=/tmp/Ultralytics
 
-# 7. Start the application
+# 10. Start application
 CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]