trinity-embedding / Dockerfile
melonceo's picture
Update Dockerfile
8e841fa verified
raw
history blame contribute delete
660 Bytes
# 1. κ°€λ²Όμš΄ Python 이미지 μ‚¬μš©
FROM python:3.10-slim
# 2. μž‘μ—… 디렉토리 μ„€μ • 및 ν™˜κ²½ λ³€μˆ˜
WORKDIR /app
ENV HOME=/app
ENV PORT=7860
# 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"]