Update Dockerfile

Dockerfile

@@ -1,55 +1,28 @@
 FROM ubuntu:latest
 
-#RUN apt-get update && apt-get install -y whoami
-
-RUN whoami
-
-# Ajouter un utilisateur non-root pour exécuter l'application (recommandé pour des raisons de sécurité)
-
-
-RUN apt update &&  apt install curl -y
+# Install necessary packages
+RUN apt-get update && apt-get install -y curl
 
+# Install Ollama
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
+# Set environment variable for Ollama host
 ENV OLLAMA_HOST=0.0.0.0
 
-RUN cut -d: -f1,3 /etc/passwd
-
-RUN userdel ubuntu && useradd -m -u 1000 appuser
-
-#RUN chown -R appuser:appuser /home/appuser
-
-#USER appuser
-
-#RUN useradd -m -u 1000 appuser 
-#&& chown -R appuser:appuser /home/appuser
-
-
-# Create the directory and give appropriate permissions
+# Create directory for Ollama and set permissions
 RUN mkdir -p /.ollama && chmod 777 /.ollama
-#RUN mkdir -p /home/appuser/.ollama && chmod 777 /home/appuser/.ollama
-#RUN mkdir -p /home/appuser/.ollama/models && 
-
-#USER appuser
-
-WORKDIR /.ollama
-
 
-#COPY --chown=appuser model_llm_local/model_ollama3 /usr/share/ollama/.ollama/models
-COPY --chown=appuser model_llm_local/model_ollama4 /usr/share/ollama/.ollama/models
+# Create directory for models and set permissions
+RUN mkdir -p /usr/share/ollama/.ollama/models && chmod -R 777 /usr/share/ollama/.ollama/models
 
-#RUN chmod 777 /home/appuser/.ollama/models
-# Copy the entry point script
-#COPY --chown=appuser entrypoint.sh /entrypoint.sh
-#RUN chmod +x /entrypoint.sh
+# Pull the model during the build process
+RUN ollama pull noahabebe/baymax
 
-# Set the entry point script as the default command
-#ENTRYPOINT ["/entrypoint.sh"]
-CMD ollama serve
-
-# Set the model as an environment variable (this can be overridden)
-#ENV model="nomic-embed-text","yayarun/mixtral_erbot"
+# Set Ollama models environment variable
 ENV OLLAMA_MODELS="/usr/share/ollama/.ollama/models"
 
 # Expose the server port
 EXPOSE 7860
+
+# Run the Ollama server
+CMD ollama serve