ani / Dockerfile
aniudupa's picture
Update Dockerfile
df3738c verified
raw
history blame
891 Bytes
# Read the doc: https://huggingface.co/docs/hub/spaces-sdks-docker
# you will also find guides on how best to write your Dockerfile
FROM python:3.9-slim
RUN useradd -m -u 1000 user
USER user
ENV PATH="/home/user/.local/bin:$PATH"
ENV HF_HOME=/home/user/huggingface
ENV TRANSFORMERS_CACHE=/home/user/huggingface/transformers
ENV SENTENCE_TRANSFORMERS_HOME=/home/user/sentence_transformers
ENV TORCH_HOME=/home/user/torch
WORKDIR /app
COPY --chown=user ./requirements.txt requirements.txt
RUN pip install --no-cache-dir --upgrade -r requirements.txt
RUN pip install --upgrade langchain pydantic transformers
RUN pip install -U langchain-huggingface
# Step 6: Install faiss-gpu separately (if needed)
RUN pip install --no-cache-dir faiss-gpu
RUN chmod -R 777 /home/user
EXPOSE 7860
COPY --chown=user . /app
CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860", "--reload"]