Update Dockerfile

Dockerfile

@@ -17,8 +17,20 @@ RUN apt-get update && apt-get install -y \
 # 수정된 명령어 (install.sh 스크립트 사용)
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Ollama 모델 다운로드 (gemma2:9b 모델 사용)
-RUN /usr/local/bin/ollama pull gemma2:9b
+# (선택) 모델 캐시 경로 명시 – 기본은 /root/.ollama
+ENV OLLAMA_MODELS=/root/.ollama
+
+# 빌드 중 잠깐 서버를 띄워서 모델만 미리 받아두고 종료
+RUN set -eux; \
+    ollama serve & pid=$!; \
+    i=0; \
+    until curl -fsS http://127.0.0.1:11434/api/tags >/dev/null 2>&1; do \
+      i=$((i+1)); \
+      [ "$i" -lt 60 ] || { echo "Ollama not ready"; kill "$pid"; exit 1; }; \
+      sleep 1; \
+    done; \
+    ollama pull gemma2:9b; \
+    kill "$pid" || true
 
 # 3. 작업 디렉터리 설정
 WORKDIR /app