Update Dockerfile

Dockerfile

@@ -16,12 +16,12 @@ USER ollama-user
 # Set Ollama to listen on all network interfaces
 ENV OLLAMA_HOST=0.0.0.0:7860
 
-# Pull the desired model during the build
-RUN ollama pull steamdj/llama3.1-cpu-only
-
-# Expose the default port (7860)
+# Expose the default port
 EXPOSE 7860
 
-# Run Ollama serve in the background
-ENTRYPOINT ["/bin/ollama"]
-CMD ["serve"]
+# Copy the entrypoint script
+COPY entrypoint.sh /usr/local/bin/entrypoint.sh
+RUN chmod +x /usr/local/bin/entrypoint.sh
+
+# Use the custom entrypoint script
+ENTRYPOINT ["/usr/local/bin/entrypoint.sh"]