Update Dockerfile

Dockerfile

@@ -1,22 +1,35 @@
 FROM python:3.10-slim-buster
-RUN apt-get update &&  apt-get install curl -y
+FROM python:3.10-slim-buster
+
+# Install curl
+RUN apt-get update && apt-get install -y curl
+# Install ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
+# Set environment variable
 ENV OLLAMA_HOST=0.0.0.0
-# Create the directory and give appropriate permissions
+# Create the directory and set permissions
 RUN mkdir -p /app/.ollama && chmod 777 /app/.ollama
+# Create a new user and group
+RUN groupadd -r app && useradd -r -g app app
+# Change ownership of the directory
+RUN chown -R app:app /app/.ollama
+# Switch to the new user
 USER app
+# Set working directory
 WORKDIR /app/.ollama
-#Copy dossier de models
-#COPY --chown=app models /.ollama
-RUN chmod 777 /app/.ollama/models
+# Copy models directory (uncomment if you have a models directory to copy)
+# COPY --chown=app:app models /app/.ollama
+# Ensure the models directory exists before changing permissions
+RUN mkdir -p /app/.ollama/models && chmod 777 /app/.ollama/models
 # Copy the entry point script
 COPY start.sh /app/start.sh
+# Make the entry point script executable
 RUN chmod +x /app/start.sh
-# Set the entry point script as the default command
-CMD ["/app/start.sh"]
-#ENV OLLAMA_MODELS="/home/app/.ollama/models"
 # Expose the server port
 EXPOSE 7860
+# Set the entry point script as the default command
+CMD ["/app/start.sh"]
+USER root 
 
 
 # Set the working directory