Update Dockerfile

Dockerfile

@@ -1,3 +1,4 @@
+# Dockerfile
 FROM python:3.11-slim
 
 # Create a non-root user
@@ -29,12 +30,28 @@ RUN chown -R ollama:ollama /app
 # Switch to ollama user
 USER ollama
 
-# Create a startup script
+# Create a startup script with model pulling
 RUN echo '#!/bin/bash\n\
 export OLLAMA_MODELS=/home/ollama/.ollama\n\
 export OLLAMA_HOST=0.0.0.0\n\
+echo "Starting Ollama server..."\n\
 ollama serve &\n\
 sleep 15\n\
+echo "Pulling model gguf-org/gemma-3-270m-gguf:Q5_K_S..."\n\
+for attempt in {1..3}; do\n\
+    if ollama pull hf.co/gguf-org/gemma-3-270m-gguf:Q5_K_S; then\n\
+        echo "Model pulled successfully"\n\
+        break\n\
+    else\n\
+        echo "Attempt $attempt: Failed to pull model, retrying in 10 seconds..."\n\
+        sleep 10\n\
+    fi\n\
+done\n\
+if [ $attempt -eq 3 ]; then\n\
+    echo "Error: Failed to pull model after 3 attempts"\n\
+    exit 1\n\
+fi\n\
+echo "Starting Gunicorn server..."\n\
 python3 -m gunicorn --bind 0.0.0.0:7860 --workers 1 --timeout 120 app:app' > /app/start.sh && \
     chmod +x /app/start.sh
 
@@ -46,4 +63,4 @@ HEALTHCHECK --interval=30s --timeout=10s --start-period=5s --retries=3 \
     CMD curl -f http://localhost:7860/health || exit 1
 
 # Run the startup script
-CMD ["/app/start.sh"]
+CMD ["/app/start.sh"]