up

Dockerfile

@@ -27,7 +27,10 @@ RUN apt-get update && apt-get install -y \
 RUN python -m pip install --upgrade pip
 RUN pip install --no-cache-dir -r requirements.txt
 
+# Install Ollama
+RUN curl -fsSL https://ollama.com/install.sh | sh
 
+RUN which ollama
 # Expose the port the application uses (replace 11434 with the actual port)
 EXPOSE 11434
 

init.sh

@@ -1,15 +1,5 @@
 #!/bin/bash
-# Install Ollama
-curl -fsSL https://ollama.com/install.sh | sh
-
-# Start Ollama in the background.
-ollama serve &
-# Record Process ID.
-pid=$!
-
-# Pause for Ollama to start.
-sleep 5
-
+which ollama
 echo "🔴 Retrieve LLAMA3 model..."
 ollama pull llama3
 echo "🟢 Done!"