bumblebee-tools / Dockerfile
jonatanklosko's picture
Add pytorch to avoid runtime warning
61262c2
raw
history blame contribute delete
835 Bytes
FROM ghcr.io/livebook-dev/livebook:latest
ENV LIVEBOOK_APP_SERVICE_NAME "🐳 Hugging Face - $SPACE_TITLE"
ENV LIVEBOOK_APP_SERVICE_URL "https://huggingface.co/spaces/$SPACE_AUTHOR_NAME/$SPACE_REPO_NAME"
ENV LIVEBOOK_UPDATE_INSTRUCTIONS_URL "https://livebook.dev"
ENV LIVEBOOK_WITHIN_IFRAME "true"
ENV LIVEBOOK_APPS_PATH "/public-apps"
ENV LIVEBOOK_APPS_PATH_WARMUP "manual"
ENV LIVEBOOK_DATA_PATH "/data"
ENV LIVEBOOK_PORT 7860
RUN apt-get update && apt-get install -y python3 python3-pip python-is-python3
RUN pip --no-cache-dir install --break-system-packages transformers torch sentencepiece protobuf
EXPOSE 7860
RUN mkdir -p /data
RUN chmod 777 /data
# The Space container runs with user ID 1000
RUN useradd -m -u 1000 user
ENV HOME=/home/user
USER user
COPY --chown=user public-apps/ /public-apps
RUN /app/bin/warmup_apps