Create Dockerfile

Dockerfile

@@ -0,0 +1,46 @@
+FROM python:3.10-slim
+
+WORKDIR /app
+
+# system packages
+RUN apt-get update && apt-get install -y \
+    git \
+    git-lfs \
+    ffmpeg \
+    libsm6 \
+    libxext6 \
+    cmake \
+    rsync \
+    libgl1 \
+    build-essential \
+    && rm -rf /var/lib/apt/lists/* \
+    && git lfs install
+
+# upgrade pip
+RUN pip install --no-cache-dir -U pip
+
+# python utilities
+RUN pip install --no-cache-dir \
+    datasets \
+    "huggingface-hub>=0.30" \
+    "hf-transfer>=0.1.4" \
+    "protobuf<4" \
+    "click<8.1" \
+    "uvicorn>=0.14.0" \
+    "websockets>=10.4" \
+    spaces \
+    "gradio[oauth,mcp]==6.9.0"
+
+# torch を先に入れる
+RUN pip install --no-cache-dir torch==2.3.1
+
+# requirements
+COPY requirements.txt .
+
+# flash-attn を含む残り
+RUN pip install --no-cache-dir -r requirements.txt
+
+# source
+COPY . .
+
+CMD ["python", "app.py"]