Initial commit

Dockerfile

@@ -23,8 +23,9 @@ RUN pip install --no-cache-dir torch==2.0.1+cpu -f https://download.pytorch.org/
 COPY . /app/
 
 # ----------- 모델 다운로드 -----------
+# Hugging Face Access Token (from Space Secret)
+ENV HF_TOKEN=$(cat /run/secrets/HF_TOKEN)
 # transformers 모델 다운로드 후 이미지에 포함
-
 RUN python /app/models/download_models.py
 
 

models/download_models.py

@@ -1,5 +1,7 @@
-from transformers import AutoTokenizer, AutoModel
 import os
+from transformers import AutoTokenizer, AutoModel
+
+token = os.getenv("HF_TOKEN")
 
 models = {
     'emotion': ('tae898/emoberta-base-ko', './models/emotion-classification-model'),
@@ -11,5 +13,5 @@ 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)
+    AutoModel.from_pretrained(name, cache_dir=path, use_auth_token=token)
+    AutoTokenizer.from_pretrained(name, cache_dir=path, use_auth_token=token)