Dockerfile

# Use an official Python runtime as a parent image
FROM python:3.11.1

# Set the working directory in the container
WORKDIR /app

# Install poetry
# RUN pip3 install poetry==1.7.1

# Copy the current directory contents into the container at /usr/src/app
COPY . .

# Install dependencies
# RUN poetry config virtualenvs.create false \
#     && poetry install --no-interaction --no-ansi
# Streamlit must be installed separately. Potentially this will cause an issue with dependencies in the future, but it's the only way it works.
# RUN pip3 install streamlit  

#  Install dependencies
RUN pip3 install -r requirements.txt

# Make a port available to the world outside this container
# The EXPOSE instruction informs Docker that the container listens on the specified network ports at runtime. Your container needs to listen to Streamlit’s (default) port 8501.
EXPOSE 8501

# The HEALTHCHECK instruction tells Docker how to test a container to check that it is still working. Your container needs to listen to Streamlit’s (default) port 8501:
HEALTHCHECK CMD curl --fail http://localhost:8501/_stcore/health

# Run the command inside your image filesystem.
CMD ["python", "train_llm.py"]

# Execute with:
# docker build -t <image_name> .    
# docker run -p 8501:8501 <image_name>