Update Dockerfile

Dockerfile

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