Update Dockerfile

Dockerfile

@@ -1,11 +1,21 @@
-# 1. 최신 안정 버전 이미지 사용
-FROM ghcr.io/huggingface/text-embeddings-inference:cpu-1.5
+# 1. 가벼운 Python 이미지 사용
+FROM python:3.10-slim
 
-# 2. 환경 변수로 모델 경로 고정
-ENV MODEL_ID=nlpai-lab/KURE-v1
+# 2. 작업 디렉토리 설정 및 환경 변수
+WORKDIR /app
+ENV HOME=/app
 ENV PORT=7860
-ENV HUGGINGFACE_HUB_CACHE=/data
 
-# 3. 모델을 명시적으로 지정하여 실행
-# --model-id를 명확하게 전달하고, CPU 환경에 맞는 설정을 추가합니다.
-ENTRYPOINT ["text-embeddings-router", "--model-id", "nlpai-lab/KURE-v1", "--port", "7860", "--max-client-batch-size", "32"]
+# 3. 필수 패키지 설치
+# sentence-transformers: 모델 실행용
+# fastapi, uvicorn: API 서버 구축용
+RUN pip install --no-cache-dir sentence-transformers fastapi uvicorn
+
+# 4. 모델 미리 다운로드 (빌드 시점에 다운로드하여 실행 속도 향상)
+RUN python -c "from sentence_transformers import SentenceTransformer; SentenceTransformer('nlpai-lab/KURE-v1')"
+
+# 5. 소스 코드 복사
+COPY . .
+
+# 6. 서버 실행
+CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]