rag-visualizer / Dockerfile
Vasanth6's picture
Replace Gemma embedding model with BAAI/bge-small-en-v1.5 and cache at build time
1142544
Raw
History Blame Contribute Delete
1.19 kB
FROM python:3.11-slim
WORKDIR /app
ENV HF_HOME=/app/.cache/huggingface
ENV NLTK_DATA=/app/nltk_data
# Install system deps for numpy/umap/llama-cpp
RUN apt-get update && apt-get install -y --no-install-recommends \
build-essential gcc g++ && \
rm -rf /var/lib/apt/lists/*
COPY pyproject.toml .
RUN pip install --no-cache-dir .
# Pre-download NLTK data
RUN mkdir -p /app/nltk_data && \
python -c "import nltk; nltk.download('punkt', download_dir='/app/nltk_data'); nltk.download('punkt_tab', download_dir='/app/nltk_data'); nltk.download('stopwords', download_dir='/app/nltk_data')"
# Pre-download public HF embedding and LLM models
RUN python -c "from huggingface_hub import snapshot_download; \
snapshot_download(repo_id='nomic-ai/nomic-embed-text-v1.5'); \
snapshot_download(repo_id='C10X/Qwen3-Embedding-TurboX.v2'); \
snapshot_download(repo_id='BAAI/bge-small-en-v1.5'); \
snapshot_download(repo_id='Qwen/Qwen2.5-0.5B-Instruct')"
COPY . .
RUN chmod -R 777 /app
EXPOSE 7860
# Force unbuffered Python output so you can actually see error logs in HF Spaces
CMD ["python", "-u", "-m", "uvicorn", "backend.main:app", "--host", "0.0.0.0", "--port", "7860"]