Spaces:
Sleeping
Sleeping
File size: 986 Bytes
0786d00 c2ee06b e5996cb b92d070 0786d00 b92d070 ed04c7a b92d070 0786d00 b92d070 e1911b7 0786d00 b92d070 8d4d615 b92d070 ed04c7a 0786d00 b92d070 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 |
# Define global args
ARG MODEL="BAAI/bge-base-en-v1.5"
FROM debian:bullseye-slim AS build-image
# Include global args in this stage of the build
ARG MODEL
ENV MODEL=${MODEL}
COPY ./download.sh ./
# Install build dependencies
RUN apt-get update && \
apt-get install -y git-lfs
RUN chmod +x *.sh && \
./download.sh && \
rm *.sh
# Stage 3 - final runtime image
# Grab a fresh copy of the Python image
FROM python:3.11-slim
# Include global args in this stage of the build
ARG MODEL
ENV MODEL=${MODEL}
ENV NORMALIZE_EMBEDDINGS=1
ENV TRANSFORMERS_CACHE="/tmp/transformers_cache"
# Set environment variable for the host
ENV HOST=0.0.0.0
ENV PORT=7860
COPY --from=build-image ${MODEL} ${MODEL}
COPY ./main.py ./
COPY ./start_server.sh ./
COPY ./index.html ./
RUN pip install --no-cache-dir open-text-embeddings[server] && \
chmod +x ./start_server.sh
# Expose a port for the server
EXPOSE ${PORT}
# Run the server start script
CMD ["/bin/sh", "./start_server.sh"]
|