Update Dockerfile

Dockerfile

@@ -8,7 +8,7 @@ ENV DEBIAN_FRONTEND=noninteractive \
     TORCH_CUDA_ARCH_LIST="7.5;8.0;8.6;8.9" \
     CMAKE_BUILD_PARALLEL_LEVEL=8
 
-# Python 3.11 + headers + toolchain + Cairo dev (for svglib/rlpycairo)
+# Python 3.11 + headers + toolchain + Cairo dev
 RUN apt-get update && apt-get install -y --no-install-recommends \
     python3.11 python3.11-venv python3.11-dev python3-pip \
     git git-lfs build-essential ninja-build cmake pkg-config \
@@ -16,29 +16,29 @@ RUN apt-get update && apt-get install -y --no-install-recommends \
     libcairo2-dev libpango1.0-dev libffi-dev \
  && rm -rf /var/lib/apt/lists/* && git lfs install \
  && ln -sf /usr/bin/python3.11 /usr/bin/python \
- && python -m pip install -U pip wheel setuptools packaging pybind11
+ && python -m pip install -U pip wheel setuptools packaging pybind11 \
+ && python -m pip install -U psutil
 
 WORKDIR /app
 COPY requirements.txt /app/requirements.txt
 
-# 1) Install Torch (CUDA 12.1) first so flash-attn can find headers/libs
+# 1) Torch (CUDA 12.1) first
 RUN pip install --index-url https://download.pytorch.org/whl/cu121 \
     torch==2.5.1 torchvision==0.20.1
 
-# 2) Build/install flash-attn against existing Torch/CUDA
+# 2) flash-attn against existing Torch/CUDA (needs psutil present)
 RUN pip install --no-build-isolation --no-cache-dir flash-attn==2.7.3
 
-# 3) Pre-install Cairo stack from wheels to avoid building pycairo
-#    (if wheels exist for cp311/manylinux, this short-circuits Meson)
+# 3) Pre-install Cairo Python wheels (avoid building pycairo)
 RUN pip install --no-cache-dir pycairo==1.26.0 rlpycairo==0.4.0 svglib==1.6.0
 
-# 4) Rest of your deps (no torch here)
+# 4) Rest of deps (no torch here)
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 5) Install StarVector package last (now its extras resolve cleanly)
+# 5) StarVector package last
 RUN pip install --no-cache-dir git+https://github.com/joanrod/star-vector.git@main
 
-# 6) Copy app code after deps (better caching)
+# 6) App code
 COPY . /app
 
 ENV PORT=7860