Spaces:
Sleeping
Sleeping
Initial commit
Browse files- .dockerignore +0 -9
- Dockerfile +12 -6
.dockerignore
CHANGED
|
@@ -20,12 +20,3 @@ __pycache__/
|
|
| 20 |
# ํ
์คํธ ์ฝ๋
|
| 21 |
tests/
|
| 22 |
|
| 23 |
-
# ๋ชจ๋ธ ์บ์ (์ ํ์ ์ผ๋ก ์ ์ง)
|
| 24 |
-
models/emotion-classification-model/*
|
| 25 |
-
!models/emotion-classification-model/.gitkeep
|
| 26 |
-
|
| 27 |
-
models/fallback-npc-model/*
|
| 28 |
-
!models/fallback-npc-model/.gitkeep
|
| 29 |
-
|
| 30 |
-
models/sentence-embedder/*
|
| 31 |
-
!models/sentence-embedder/.gitkeep
|
|
|
|
| 20 |
# ํ
์คํธ ์ฝ๋
|
| 21 |
tests/
|
| 22 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
Dockerfile
CHANGED
|
@@ -23,12 +23,18 @@ RUN pip install --no-cache-dir torch==2.0.1+cpu -f https://download.pytorch.org/
|
|
| 23 |
COPY . /app/
|
| 24 |
|
| 25 |
# ----------- ๊ถํ ์ค์ ๋ฐ ์ด๊ธฐํ -----------
|
| 26 |
-
|
| 27 |
-
|
| 28 |
-
|
| 29 |
-
|
| 30 |
-
|
| 31 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 32 |
|
| 33 |
ENV EMOTION_MODEL_NAME=tae898/emoberta-base-ko
|
| 34 |
ENV FALLBACK_MODEL_NAME=skt/ko-gpt-trinity-1.2B-v0.5
|
|
|
|
| 23 |
COPY . /app/
|
| 24 |
|
| 25 |
# ----------- ๊ถํ ์ค์ ๋ฐ ์ด๊ธฐํ -----------
|
| 26 |
+
# ๋๋ ํ ๋ฆฌ ์์ฑ
|
| 27 |
+
RUN mkdir -p /app/models/emotion-classification-model \
|
| 28 |
+
&& mkdir -p /app/models/fallback-npc-model \
|
| 29 |
+
&& mkdir -p /app/models/sentence-embedder \
|
| 30 |
+
&& mkdir -p /app/rag/chroma_DB \
|
| 31 |
+
&& mkdir -p /app/.cache/huggingface/hub
|
| 32 |
+
|
| 33 |
+
# ๊ถํ ๋ถ์ฌ
|
| 34 |
+
RUN chmod -R 777 /app
|
| 35 |
+
|
| 36 |
+
# ENV ์ค์
|
| 37 |
+
ENV TRANSFORMERS_CACHE=/app/.cache/huggingface/hub
|
| 38 |
|
| 39 |
ENV EMOTION_MODEL_NAME=tae898/emoberta-base-ko
|
| 40 |
ENV FALLBACK_MODEL_NAME=skt/ko-gpt-trinity-1.2B-v0.5
|