textClassifierAPI / Dockerfile
hiddenFront's picture
Update Dockerfile
e2b01db verified
# Python 3.9 Slim ์ด๋ฏธ์ง€๋ฅผ ๊ธฐ๋ฐ˜์œผ๋กœ ์‚ฌ์šฉํ•ฉ๋‹ˆ๋‹ค.
FROM python:3.9-slim
# ํ•„์ˆ˜ ํŒจํ‚ค์ง€ ๋ฐ ๋นŒ๋“œ ๋„๊ตฌ ์„ค์น˜
RUN apt-get update && apt-get install -y --no-install-recommends \
git \
build-essential \
curl \
libffi-dev \
libssl-dev \
&& rm -rf /var/lib/apt/lists/*
# Rust ์ปดํŒŒ์ผ๋Ÿฌ ์„ค์น˜
RUN curl --proto '=https' --tlsv1.2 -sSf https://sh.rustup.rs | sh -s -- -y --default-toolchain stable
ENV PATH="/root/.cargo/bin:${PATH}"
# --- Hugging Face ์บ์‹œ ํ™˜๊ฒฝ ๋ณ€์ˆ˜ ์„ค์ • ๋ฐ ๋””๋ ‰ํ† ๋ฆฌ ์ƒ์„ฑ ---
# Hugging Face ๋ผ์ด๋ธŒ๋Ÿฌ๋ฆฌ๊ฐ€ ์บ์‹œ ํŒŒ์ผ์„ ์“ฐ๊ธฐ ๊ฐ€๋Šฅํ•œ /tmp ๋””๋ ‰ํ† ๋ฆฌ์— ์ €์žฅํ•˜๋„๋ก ์„ค์ •ํ•ฉ๋‹ˆ๋‹ค.
# ๋””๋ ‰ํ† ๋ฆฌ๋ฅผ ๋ฏธ๋ฆฌ ์ƒ์„ฑํ•˜๊ณ  ๋ชจ๋“  ์‚ฌ์šฉ์ž์—๊ฒŒ ์“ฐ๊ธฐ ๊ถŒํ•œ์„ ๋ถ€์—ฌํ•ฉ๋‹ˆ๋‹ค.
ENV TRANSFORMERS_CACHE="/tmp/huggingface_cache"
ENV HF_HOME="/tmp/huggingface_cache"
RUN mkdir -p /tmp/huggingface_cache && chmod -R 777 /tmp/huggingface_cache
# ์ž‘์—… ๋””๋ ‰ํ† ๋ฆฌ ์„ค์ •
WORKDIR /app
# ์ „์ฒด ์ฝ”๋“œ ๋ณต์‚ฌ
COPY . .
# Python ํŒจํ‚ค์ง€ ์„ค์น˜
RUN pip install --no-cache-dir --upgrade pip
# ๋ชจ๋“  ํŒจํ‚ค์ง€๋ฅผ ํ•œ ๋ฒˆ์— ์„ค์น˜ํ•ฉ๋‹ˆ๋‹ค.
RUN pip install --no-cache-dir \
fastapi \
uvicorn \
torch==1.13.1 \
transformers==4.30.2 \
tokenizers==0.13.3 \
sentencepiece \
numpy==1.23.5 \
protobuf==3.20.3 \
psutil \
gluonnlp==0.10.0 \
mxnet-mkl==1.6.0 \
huggingface_hub
# CMD๋Š” app.py๋ฅผ ์‹คํ–‰ํ•˜๋„๋ก ์„ค์ •๋˜์–ด ์žˆ์Šต๋‹ˆ๋‹ค.
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]