File size: 577 Bytes
490e475
 
 
 
d1033d4
 
 
490e475
d1033d4
 
 
 
490e475
d1033d4
490e475
d1033d4
490e475
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
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"]