.

Dockerfile

@@ -19,7 +19,12 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-COPY --chown=user . /app
+# --- [수정] 모델 미리 다운로드 (캐싱) ---
+# 소스코드 복사 전에 모델 다운로드를 수행하여 Docker 레이어 캐시 활용
+COPY --chown=user ./download_models.py download_models.py
+RUN python download_models.py
+
+COPY --chown=user . .
 
 #서버 실행 명령: 컨테이너가 시작될 때 FastAPI 서버를 실행하도록 명령을 정의합니다. 
 #app.py 파일의 app 객체를 7860 포트로 열어줍니다. 

download_models.py

@@ -0,0 +1,18 @@
+# download_models.py
+import os
+from huggingface_hub import hf_hub_download
+import torchvision.models as models
+
+def download_all():
+    print("Download ONNX models...")
+    model_id = "onnx-community/embeddinggemma-300m-ONNX"
+    # 토큰은 공개 모델이면 필요 없으나, 필요시 빌드 아규먼트로 처리해야 함
+    hf_hub_download(model_id, subfolder="onnx", filename="model.onnx")
+    hf_hub_download(model_id, subfolder="onnx", filename="model.onnx_data")
+    
+    print("Download Vision models...")
+    models.efficientnet_b1(weights='IMAGENET1K_V1')
+    print("Download Complete.")
+
+if __name__ == "__main__":
+    download_all()