FROM python:3.9

WORKDIR /code

COPY ./requirements.txt /code/requirements.txt

RUN pip install --no-cache-dir --upgrade -r /code/requirements.txt

RUN git clone https://github.com/the-crypt-keeper/llm-visualizer.git /code/llm-visualizer

WORKDIR /code/llm-visualizer

ENV TRANSFORMERS_CACHE=/tmp/

CMD ["streamlit", "run", "app.py", "--server.address", "0.0.0.0", "--server.port", "7860"]