Update Dockerfile

Dockerfile

@@ -13,9 +13,15 @@ RUN mkdir -p $HOME/.ollama && chown -R ollama-user:ollama-user $HOME/.ollama
 # Switch to the new user
 USER ollama-user
 
+# Set Ollama to listen on all network interfaces
 ENV OLLAMA_HOST=0.0.0.0:7860
 
-RUN ollama serve && ollama pull steamdj/llama3.1-cpu-only
+# Pull the desired model during the build
+RUN ollama pull steamdj/llama3.1-cpu-only
 
-# ENTRYPOINT ["/bin/ollama"]
-# CMD ["serve"]
+# Expose the default port (7860)
+EXPOSE 7860
+
+# Run Ollama serve in the background
+ENTRYPOINT ["/bin/ollama"]
+CMD ["serve"]