Update Dockerfile

Dockerfile

@@ -1,7 +1,7 @@
 # Builder stage
 FROM ubuntu:latest
 
-RUN apt update &&  apt install curl -y
+RUN apt update && apt install curl -y
 
 RUN curl https://ollama.ai/install.sh | sh
 
@@ -10,11 +10,16 @@ RUN mkdir -p /.ollama && chmod 777 /.ollama
 
 WORKDIR /.ollama
 
-# Command to run the application
-CMD ollama serve 
+# Copy the entry point script
+COPY entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
 
-# Command to pull model
-CMD ollama pull llama2
+# Set the entry point script as the default command
+ENTRYPOINT ["/entrypoint.sh"]
+CMD ["ollama", "serve"]
+
+# Set the model as an environment variable (this can be overridden)
+ENV model=llama2
 
 # Expose the server port
-EXPOSE 7860
+EXPOSE 7860