Update Dockerfile

Dockerfile

@@ -12,10 +12,10 @@ RUN sed -i 's/main/main contrib non-free/' /etc/apt/sources.list && \
 
 WORKDIR /app
 
-# Install Python dependencies in optimal order
-RUN pip install --no-cache-dir --upgrade pip setuptools wheel
+# Install core dependencies first
+RUN pip install --no-cache-dir --upgrade pip setuptools wheel packaging
 
-# Install CPU-only PyTorch and compatible NumPy first
+# Install CPU-only PyTorch and compatible NumPy
 RUN pip install --no-cache-dir \
     "numpy<2" \
     torch==2.1.2+cpu \
@@ -23,12 +23,24 @@ RUN pip install --no-cache-dir \
     torchaudio==2.1.2+cpu \
     --index-url https://download.pytorch.org/whl/cpu
 
-# Install other requirements
+# Copy requirements and install with flash_attn workaround
 COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
-
-# Remove flash-attn if not needed (GPU-only package)
-RUN pip uninstall -y flash-attn || true
+RUN pip install --no-cache-dir -r requirements.txt --no-deps && \
+    pip install --no-cache-dir \
+    accelerate \
+    transformers==4.36.2 \
+    timm==0.9.12 \
+    einops==0.7.0
+
+# Install remaining packages
+RUN pip install --no-cache-dir \
+    fastapi \
+    uvicorn \
+    opencv-python-headless \
+    pathlib \
+    huggingface_hub \
+    pillow \
+    rarfile
 
 # Copy application code
 COPY . .