# Use Ubuntu 22.04 as the base image
FROM ubuntu:22.04

# Install Python 3 and pip
RUN apt-get update && \
    apt-get install -y python3 python3-pip && \
    apt-get clean && \
    rm -rf /var/lib/apt/lists/*

# Upgrade pip and install Python packages
RUN pip3 install --upgrade pip
RUN pip3 install transformers
RUN pip3 install "fschat[model_worker,webui]"
RUN pip install accelerate pandas numpy


# Add the entrypoint script
ADD entrypoint.sh /entrypoint.sh

# Set the script as the entry point
ENTRYPOINT ["/entrypoint.sh"]