m97j commited on
Commit
b635cd2
ยท
1 Parent(s): 54f8424

Initial commit

Browse files
Files changed (2) hide show
  1. .dockerignore +0 -9
  2. 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
- RUN mkdir -p /app/rag/chroma_DB && chmod -R 777 /app/rag/chroma_DB
27
-
28
- RUN mkdir -p /app/models/emotion-classification-model && chmod -R 777 /app/models
29
- RUN mkdir -p /app/models/fallback-npc-model && chmod -R 777 /app/models
30
- RUN mkdir -p /app/models/sentence-embedder && chmod -R 777 /app/models
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