Update entrypoint.sh

entrypoint.sh

@@ -1,6 +1,13 @@
 #!/bin/bash
-# Start Ollama in the background
-ollama serve 
+
+# Define Ollama directory within the container's working directory
+OLLAMA_DIR="./ollama"
+
+# Create the Ollama directory if it does not exist
+mkdir -p ${OLLAMA_DIR}
+
+# Start Ollama in the background, specifying the directory for its data
+ollama serve --dir ${OLLAMA_DIR}
 
 ollama run llama2-uncensored &