Initial commit

Dockerfile

@@ -24,19 +24,9 @@ COPY . /app/
 
 # ----------- 모델 다운로드 -----------
 # transformers 모델 다운로드 후 이미지에 포함
-RUN python -c "from transformers import AutoTokenizer, AutoModel; \
-    models = { \
-        'emotion': ('tae898/emoberta-base-ko', './models/emotion-classification-model'), \
-        'fallback': ('skt/ko-gpt-trinity-1.2B-v0.5', './models/fallback-npc-model'), \
-        'embedder': ('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2', './models/sentence-embedder') \
-    }; \
-    import os; \
-    os.makedirs('./models/emotion-classification-model', exist_ok=True); \
-    os.makedirs('./models/fallback-npc-model', exist_ok=True); \
-    os.makedirs('./models/sentence-embedder', exist_ok=True); \
-    for key, (name, path) in models.items(): \
-        AutoModel.from_pretrained(name, cache_dir=path); \
-        AutoTokenizer.from_pretrained(name, cache_dir=path)"
+
+RUN python /app/models/download_models.py
+
 
 # ----------- 환경 변수 설정 -----------
 ENV EMOTION_MODEL_DIR=/app/models/emotion-classification-model

models/download_models.py

@@ -0,0 +1,15 @@
+from transformers import AutoTokenizer, AutoModel
+import os
+
+models = {
+    'emotion': ('tae898/emoberta-base-ko', './models/emotion-classification-model'),
+    'fallback': ('skt/ko-gpt-trinity-1.2B-v0.5', './models/fallback-npc-model'),
+    'embedder': ('sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2', './models/sentence-embedder')
+}
+
+for _, (_, path) in models.items():
+    os.makedirs(path, exist_ok=True)
+
+for key, (name, path) in models.items():
+    AutoModel.from_pretrained(name, cache_dir=path)
+    AutoTokenizer.from_pretrained(name, cache_dir=path)