will / Dockerfile
matt1847's picture
リファクタ: srcディレクトリ構造への移行とDocker対応
d1033d4
FROM python:3.11-slim
WORKDIR /app
# 依存関係のインストール
COPY requirements.txt .
RUN pip install --no-cache-dir -r requirements.txt
# モデルの事前ダウンロード(ビルド時にキャッシュ)
RUN python -c "from transformers import GPT2LMHeadModel, GPT2Tokenizer; \
GPT2LMHeadModel.from_pretrained('gpt2'); \
GPT2Tokenizer.from_pretrained('gpt2')"
# アプリケーションコードをコピー
COPY app.py .
COPY src/ ./src/
EXPOSE 8501
CMD ["streamlit", "run", "app.py", "--server.headless", "true", "--server.address", "0.0.0.0"]