Spaces:
Sleeping
Sleeping
# ----------- λ² μ΄μ€ μ΄λ―Έμ§ ----------- | |
FROM python:3.10-slim | |
# ----------- μμ λλ ν 리 ----------- | |
WORKDIR /app | |
# ----------- μμ€ν ν¨ν€μ§ μ€μΉ ----------- | |
RUN apt-get update && apt-get install -y --no-install-recommends \ | |
build-essential \ | |
cmake \ | |
&& rm -rf /var/lib/apt/lists/* | |
# ----------- μμ‘΄μ± μ€μΉ ----------- | |
# requirements λ¨Όμ λ³΅μ¬ β μΊμ νμ© κ°λ₯ | |
COPY requirements.txt /app/requirements.txt | |
# PyTorch CPU λ²μ μ€μΉ (GPU νμ μμ) | |
RUN pip install --no-cache-dir torch==2.0.1+cpu -f https://download.pytorch.org/whl/cpu/torch_stable.html \ | |
&& pip install --no-cache-dir -r /app/requirements.txt | |
# ----------- μ½λ λ³΅μ¬ ----------- | |
# μμ‘΄μ± μ€μΉ ν μ½λ λ³΅μ¬ β requirements λ³κ²½ μλ ν μΊμ μ¬μ¬μ© κ°λ₯ | |
COPY . /app/ | |
# ----------- κΆν μ€μ λ° μ΄κΈ°ν ----------- | |
# κΆν μ€μ | |
RUN chmod -R 777 /app | |
# λλ ν 리 μμ± | |
RUN mkdir -p /app/models/fallback-npc-model \ | |
&& mkdir -p /app/models/sentence-embedder \ | |
&& mkdir -p /app/rag/chroma_DB \ | |
&& mkdir -p /app/.cache/huggingface/hub \ | |
mkdir -p /app/.cache/sentence_transformers && chmod -R 777 /app/.cache | |
# νκ²½λ³μ μ€μ | |
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub | |
ENV SENTENCE_TRANSFORMERS_HOME=/app/.cache/sentence_transformers | |
ENV FALLBACK_MODEL_NAME=skt/ko-gpt-trinity-1.2B-v0.5 | |
ENV EMBEDDER_MODEL_NAME=sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2 | |
# ----------- ν¬νΈ μ€μ ----------- | |
EXPOSE 7860 | |
# ----------- μ€ν λͺ λ Ή ----------- | |
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"] | |