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