Update Dockerfile

Dockerfile

@@ -4,20 +4,24 @@ FROM python:3.9-slim
 # Docker 컨테이너의 작업 디렉토리를 /app으로 설정합니다.
 WORKDIR /app
 
-# **[최적화] 모델 캐시 경로를 먼저 지정합니다.**
-ENV SENTENCE_TRANSFORMERS_HOME /app/.cache
-ENV HUGGINGFACE_HUB_CACHE /app/.cache
+# Hugging Face 및 Sentence Transformers 캐시 경로 지정
+ENV SENTENCE_TRANSFORMERS_HOME=/app/.cache \
+    HUGGINGFACE_HUB_CACHE=/app/.cache \
+    HF_HOME=/app/.cache
 
-# **[최적화] 모델 다운로드 전에 캐시 폴더를 생성하고 권한을 부여합니다.**
-# 이렇게 하면 pip install 과정에서 발생하는 권한 문제를 방지할 수 있습니다.
+# 캐시 폴더 생성 및 권한 부여
 RUN mkdir -p /app/.cache && chmod -R 777 /app/.cache
 
-# requirements.txt 파일을 컨테이너의 작업 디렉토리로 복사하고, 필요한 패키지를 설치합니다.
+# requirements.txt 복사 및 패키지 설치
 COPY requirements.txt .
 RUN pip install --no-cache-dir -r requirements.txt
 
-# 현재 로컬 디렉토리의 모든 파일(.py 등)을 컨테이너로 복사합니다.
+# 앱 소스 복사
 COPY . .
 
-# 웹 서버 실행 명령어를 실행합니다.
-CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "app:app", "--bind", "0.0.0.0:7860"]
+# 빌드 시 모델 미리 다운로드
+RUN python -c "from huggingface_hub import snapshot_download; snapshot_download(repo_id='nlpai-lab/KURE-v1', local_dir='/app/.cache/nlpai-lab/KURE-v1')" \
+ && chmod -R 777 /app/.cache
+
+# Gunicorn + Uvicorn 실행
+CMD ["gunicorn", "-w", "4", "-k", "uvicorn.workers.UvicornWorker", "app:app", "--bind", "0.0.0.0:7860"]