FROM ollama/ollama
COPY ./run-ollama.sh /tmp/run-ollama.sh
WORKDIR /tmp
ARG MODEL_NAME
RUN chmod +x run-ollama.sh \
    && env INSTALL_MODEL=$MODEL_NAME ./run-ollama.sh