grouped-sampling-demo / Dockerfile
yonikremer's picture
added cache folders
0232d80
raw
history blame
376 Bytes
FROM bitnami/pytorch
WORKDIR /app
COPY ./requirements.txt /app/requirements.txt
RUN pip install --no-cache-dir -r /app/requirements.txt
RUN mkdir "/.cache/"
RUN mkdir "/.cache/huggingface/"
ENV HUGGINGFACE_HUB_CACHE="/.cache/huggingface"
RUN mkdir "/.cache/transformers/"
ENV TRANSFORMERS_CACHE="/.cache/transformers"
COPY . /app
CMD ["streamlit", "run", "/app/app.py"]