optimizations 1

.dockerignore

@@ -0,0 +1,18 @@
+.git
+.gitignore
+.gitattributes
+__pycache__
+*.pyc
+*.pyo
+*.egg-info
+.pytest_cache
+.venv
+venv
+.DS_Store
+*.md
+.vscode
+.idea
+node_modules
+.next
+build
+dist

Dockerfile

@@ -1,26 +1,26 @@
 FROM python:3.10-slim
 
-# Install system deps needed by OpenCV
+# Install system deps needed by OpenCV (minimal set)
 RUN apt-get update && apt-get install -y --no-install-recommends \
     libglib2.0-0 \
     libsm6 \
     libxext6 \
     libxrender-dev \
     libgl1 \
-    && rm -rf /var/lib/apt/lists/*
+    && rm -rf /var/lib/apt/lists/* /tmp/* /var/tmp/*
 
 WORKDIR /app
 
-# ── Copy requirements FIRST so pip layer is cached on code-only changes ──
+# ── Install dependencies FIRST (most stable layer) ──
 COPY requirements.txt .
-
-# Install with --no-cache-dir to reduce build time
-# PyTorch CPU will still take time but this reduces disk I/O
 RUN pip install --no-cache-dir --upgrade pip setuptools wheel && \
     pip install --no-cache-dir --default-timeout=1000 -r requirements.txt
 
-# ── Copy rest of the app AFTER deps are installed ──
-COPY . .
+# ── Copy model files SECOND (slower to change than code) ──
+COPY trainedmodels/ ./trainedmodels/
+
+# ── Copy app code LAST (fastest layer, changes most often) ──
+COPY app.py inference.py README.md ./
 
 # HuggingFace Spaces requires port 7860
 EXPOSE 7860

requirements.txt

@@ -1,5 +1,4 @@
 --extra-index-url https://download.pytorch.org/whl/cpu
---extra-index-url https://pypi.org/simple
 
 torch==2.0.1+cpu
 torchvision==0.15.2+cpu
@@ -8,7 +7,9 @@ fastapi==0.111.0
 uvicorn[standard]==0.29.0
 python-multipart==0.0.9
 opencv-python-headless==4.9.0.80
-albumentations==1.4.2
 numpy==1.26.4
-matplotlib==3.8.4
 Pillow==10.3.0
+
+# Optional: Remove these if not used in inference
+# albumentations==1.4.2
+# matplotlib==3.8.4