Start Ollama server in container

Dockerfile

@@ -28,33 +28,33 @@ RUN apt-get update && \
 
 RUN curl -fsSL https://ollama.com/install.sh | sh
 ARG OLLAMA_MODEL
-COPY ollama_init.sh .
-RUN chmod +x ollama_init.sh
-RUN ./ollama_init.sh ${OLLAMA_MODEL}
 
 RUN useradd -m -u 1000 user
 USER user
 ENV HOME=/home/user \
-    PATH=/home/user/.local/bin:${PATH}
+PATH=/home/user/.local/bin:${PATH}
 WORKDIR ${HOME}/app
 
 RUN curl https://pyenv.run | bash
 ENV PATH=${HOME}/.pyenv/shims:${HOME}/.pyenv/bin:${PATH}
 ARG PYTHON_VERSION=3.10.12
 RUN pyenv install ${PYTHON_VERSION} && \
-    pyenv global ${PYTHON_VERSION} && \
-    pyenv rehash && \
-    pip install --no-cache-dir -U pip setuptools wheel && \
-    pip install packaging ninja
+pyenv global ${PYTHON_VERSION} && \
+pyenv rehash && \
+pip install --no-cache-dir -U pip setuptools wheel && \
+pip install packaging ninja
 
 COPY --chown=1000 ./requirements.txt /tmp/requirements.txt
 RUN pip install --no-cache-dir --upgrade -r /tmp/requirements.txt && \
     pip install flash-attn --no-build-isolation
 
-COPY --chown=1000 src ${HOME}/app
-
+    COPY --chown=1000 src ${HOME}/app
+    
+COPY ollama_init.sh .
+RUN sh ./ollama_init.sh ${OLLAMA_MODEL}
 EXPOSE 8501
-CMD streamlit run app.py \
+CMD ollama serve & \
+    streamlit run app.py \
     --server.headless true \
     --server.enableCORS false \
     --server.enableXsrfProtection false \