Initial commit

Dockerfile

@@ -23,8 +23,6 @@ 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,7 +1,9 @@
 import os
 from transformers import AutoTokenizer, AutoModel
 
-token = os.getenv("HF_TOKEN")
+# Hugging Face token from secret file
+with open("/run/secrets/HF_TOKEN", "r") as f:
+    token = f.read().strip()
 
 models = {
     'emotion': ('tae898/emoberta-base-ko', './models/emotion-classification-model'),
@@ -12,6 +14,6 @@ models = {
 for _, (_, path) in models.items():
     os.makedirs(path, exist_ok=True)
 
-for key, (name, path) in models.items():
+for name, path in models.values():
     AutoModel.from_pretrained(name, cache_dir=path, use_auth_token=token)
     AutoTokenizer.from_pretrained(name, cache_dir=path, use_auth_token=token)