Update Dockerfile

Dockerfile

@@ -1,28 +1,26 @@
+# Use the latest Ubuntu image as the base
 FROM ubuntu:latest
 
-# Install necessary packages
+# Update the package list and install curl
 RUN apt-get update && apt-get install -y curl
 
-# Install Ollama
+# Install Ollama using the official install script
 RUN curl -fsSL https://ollama.com/install.sh | sh
 
-# Set environment variable for Ollama host
+# Set the Ollama host environment variable
 ENV OLLAMA_HOST=0.0.0.0
 
-# Create directory for Ollama and set permissions
+# Create the Ollama directory and set permissions
 RUN mkdir -p /.ollama && chmod 777 /.ollama
 
-# Create directory for models and set permissions
+# Create the models directory and set permissions
 RUN mkdir -p /usr/share/ollama/.ollama/models && chmod -R 777 /usr/share/ollama/.ollama/models
 
-# Pull the model during the build process
-RUN ollama pull noahabebe/baymax
-
-# Set Ollama models environment variable
+# Set the Ollama models environment variable
 ENV OLLAMA_MODELS="/usr/share/ollama/.ollama/models"
 
-# Expose the server port
+# Expose the Ollama server port
 EXPOSE 7860
 
-# Run the Ollama server
-CMD ollama serve
+# Run the Ollama server and pull the Baymax model
+CMD ["ollama", "serve", "&", "sleep", "5", "&&", "ollama", "pull", "noahabebe/baymax", "&&", "wait"]