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"] | |