FROM python:3.9-slim

# 필수 패키지 설치
RUN apt-get update && apt-get install -y git

# 작업 디렉토리 설정
WORKDIR /app

# requirements 먼저 복사

# requirements 설치 (kobert_tokenizer 제외)
RUN pip install --no-cache-dir --upgrade pip \
 && pip install --no-cache-dir --upgrade \
    fastapi \
    uvicorn \
    torch==1.13.1 \
    transformers==3.0.2 \
    tokenizers==0.8.1rc1 \
    sentencepiece \
    numpy==1.23.1 \
    protobuf==3.20.3 \
    psutil \
    gluonnlp==0.10.0 \
    mxnet-mkl==1.6.0 \
 && pip install git+https://github.com/SKTBrain/KOBERT#egg=kobert_tokenizer\&subdirectory=kobert_tokenizer

# 전체 코드 복사
COPY . .

# FastAPI 실행
CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "7860"]