upd

Dockerfile

@@ -1,11 +1,26 @@
 FROM python:3.10-slim-buster
-# Set up the working directory
-WORKDIR /
-SHELL ["/bin/bash", "-o", "pipefail", "-c"]
-# Copy a custom entrypoint script into the container
-COPY ollama_start.sh /
-RUN chmod +x /ollama_start.sh
-CMD ["ollama_start.sh"]
+
+RUN apt update &&  apt install curl -y
+RUN curl -fsSL https://ollama.com/install.sh | sh
+ENV OLLAMA_HOST=0.0.0.0
+RUN useradd -m app && chown -R app:app /home/app
+# Create the directory and give appropriate permissions
+RUN mkdir -p /home/app/.ollama && chmod 777 /home/app/.ollama
+#RUN mkdir -p /home/app/.ollama/models && 
+USER app
+WORKDIR /home/app/.ollama
+#Copy dossier de models
+#COPY --chown=app models /.ollama
+#RUN chmod 777 /home/app/.ollama/models
+# Copy the entry point script
+COPY --chown=app entrypoint.sh /entrypoint.sh
+RUN chmod +x /start.sh
+# Set the entry point script as the default command
+CMD ["/start.sh"]
+#ENV OLLAMA_MODELS="/home/app/.ollama/models"
+# Expose the server port
+EXPOSE 7860
+
 
 # Set the working directory
 WORKDIR /app

start.sh

@@ -0,0 +1,21 @@
+#!/bin/bash
+
+# Starting server
+echo "Starting server"
+ollama serve &
+sleep 1
+
+# Splitting the models by comma and pulling each
+IFS=',' read -ra MODELS <<< "$model"
+for m in "${MODELS[@]}"; do
+    echo "Pulling $m"
+    ollama pull "$m"
+    sleep 5
+    ollama pull "$m"
+    # echo "Running $m"
+    # ollama run "$m"
+    # No need to sleep here unless you want to give some delay between each pull for some reason
+done
+
+# Keep the script running to prevent the container from exiting
+wait