Update Dockerfile

Dockerfile

@@ -1,9 +1,8 @@
+# Dockerfile.cpu
 FROM python:3.10-slim
 
 WORKDIR /app
-
-# system packages
-RUN apt-get update && apt-get install -y \
+RUN apt-get update && apt-get install -y --no-install-recommends \
     git \
     git-lfs \
     ffmpeg \
@@ -16,10 +15,9 @@ RUN apt-get update && apt-get install -y \
     && rm -rf /var/lib/apt/lists/* \
     && git lfs install
 
-# upgrade pip
-RUN pip install --no-cache-dir -U pip
+RUN pip install --no-cache-dir -U pip setuptools wheel
 
-# python utilities
+# HF 関連等
 RUN pip install --no-cache-dir \
     datasets \
     "huggingface-hub>=0.30" \
@@ -31,16 +29,10 @@ RUN pip install --no-cache-dir \
     spaces \
     "gradio[oauth,mcp]==6.9.0"
 
-# torch を先に入れる
-RUN pip install --no-cache-dir torch==2.3.1
-
-# requirements
-COPY requirements.txt .
+COPY requirements.txt /tmp/requirements.txt
+# requirements.txt に flash_attn が入っていたら除外しておく
+RUN sed -i '/flash_attn/d' /tmp/requirements.txt && \
+    pip install --no-cache-dir -r /tmp/requirements.txt
 
-# flash-attn を含む残り
-RUN pip install --no-cache-dir -r requirements.txt
-
-# source
 COPY . .
-
 CMD ["python", "app.py"]