Update Dockerfile

Dockerfile

@@ -1,17 +1,26 @@
-# Builder stage
-FROM ubuntu:latest
+FROM ollama/ollama AS base
 
-RUN apt update &&  apt install curl -y
+# Create the /.ollama directory and ensure it has the right permissions
+USER root
+RUN mkdir -p /.ollama && \
+    chmod -R 777 /.ollama
 
-RUN curl https://ollama.ai/install.sh | sh
+# Copy your scripts and set them up
+COPY ollama-pull.sh /ollama-pull.sh
+RUN chmod +x /ollama-pull.sh
 
-# Create the directory and give appropriate permissions
-RUN mkdir -p /.ollama && chmod 777 /.ollama
+# Switch to your application's user if it doesn't require root
+# USER appuser
 
-WORKDIR /.ollama
+ENV model="llama2"
 
-# Command to run the application
-CMD ollama serve 
+# Run your script as the last step of the build process
+RUN /ollama-pull.sh
 
-# Expose the server port
-EXPOSE 7860
+FROM base AS final-stage
+
+# Setup entrypoint and default command
+ENTRYPOINT ["/bin/ollama"]
+EXPOSE 11434
+ENV OLLAMA_HOST 0.0.0.0
+CMD ["serve"]