Update Dockerfile

Dockerfile

@@ -1,19 +1,30 @@
+# Use the latest version of the ollama image
 FROM ollama/ollama:latest
 
-RUN apt-get update && apt-get install curl -y
+# Install curl and create a user
+RUN apt-get update && \
+    apt-get install -y curl && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/* && \
+    useradd -m -u 1000 user
 
-# https://huggingface.co/docs/hub/spaces-sdks-docker-first-demo
-RUN useradd -m -u 1000 user
-USER user
+# Set environment variables
 ENV HOME=/home/user \
-	PATH=/home/user/.local/bin:$PATH \
+    PATH=/home/user/.local/bin:$PATH \
     OLLAMA_HOST=0.0.0.0 \
     OLLAMA_ORIGINS=*
 
+# Set the working directory and switch to the user
 WORKDIR $HOME/app
+USER user
 
-COPY --chown=user:user Modelfile $HOME/app/
+COPY --chown=user:user entrypoint.sh $HOME/app/
 
-RUN ollama serve & sleep 5 && ollama pull qwen2:0.5b
+# Make the entrypoint script executable
+RUN chmod +x $HOME/app/entrypoint.sh
 
+# Expose the port for the Ollama server
 EXPOSE 11434
+
+# Set the entrypoint to the entrypoint script
+ENTRYPOINT ["./entrypoint.sh"]