Dockerfile start.sh

Dockerfile

@@ -0,0 +1,14 @@
+FROM ollama/ollama:latest
+
+# Ensure external access within the container
+ENV OLLAMA_HOST=0.0.0.0
+
+# Copy a startup script
+COPY start.sh /start.sh
+RUN chmod +x /start.sh
+
+# Hugging Face Spaces typically use port 7860
+EXPOSE 7860
+ENV OLLAMA_PORT=7860
+
+ENTRYPOINT ["/bin/bash", "/start.sh"]

start.sh

@@ -0,0 +1,29 @@
+#!/bin/bash
+
+# Start Ollama server in the background
+# We bind it to port 7860 as expected by HF Spaces
+OLLAMA_HOST=0.0.0.0:7860 ollama serve &
+
+# Wait for the server to be ready
+until curl -s localhost:7860/api/tags > /dev/null; do
+  echo "Waiting for Ollama server..."
+  sleep 2
+done
+
+# Pull the cloud model using the secret environment variable
+# HF automatically injects your secret as an ENV variable
+if [ -n "$OLLAMA_API_KEY" ]; then
+    echo "Authenticating and pulling cloud model..."
+    ollama pull gpt-oss:120b-cloud
+    ollama pull deepseek-v3.2:cloud
+    ollama pull gemini-3-flash-preview:cloud
+    ollama pull minimax-m2.7:cloud
+    ollama pull gemma4:31b-cloud
+else
+    echo "Error: OLLAMA_API_KEY secret not found in Space settings."
+fi
+
+ollama ls
+
+# Keep the script running to keep the container alive
+wait