Create start.sh

start.sh

@@ -0,0 +1,29 @@
+# start.sh
+#!/bin/bash
+set -e
+LOG_FILE=/data/startup.log
+mkdir -p /data && touch $LOG_FILE && chmod 644 $LOG_FILE
+echo "Starting Ollama server at $(date)" >> $LOG_FILE
+ollama serve >> $LOG_FILE 2>&1 &
+sleep 15
+MODELS_TO_PULL="${MODELS_TO_PULL:-gemma-3-270m}"
+echo "Pulling models: $MODELS_TO_PULL" | tee -a $LOG_FILE
+IFS=',' read -ra MODEL_ARRAY <<< "$MODELS_TO_PULL"
+for model in "${MODEL_ARRAY[@]}"; do
+    echo "Pulling model $model..." | tee -a $LOG_FILE
+    for attempt in {1..3}; do
+        if ollama pull "$model" >> $LOG_FILE 2>&1; then
+            echo "Model $model pulled successfully" | tee -a $LOG_FILE
+            break
+        else
+            echo "Attempt $attempt: Failed to pull model $model, retrying in 10 seconds..." | tee -a $LOG_FILE
+            sleep 10
+        fi
+        if [ $attempt -eq 3 ]; then
+            echo "Error: Failed to pull model $model after 3 attempts" | tee -a $LOG_FILE
+            exit 1
+        fi
+    done
+done
+echo "Starting Gunicorn server at $(date)" | tee -a $LOG_FILE
+exec gunicorn --bind 0.0.0.0:7860 --workers 1 --timeout 120 --log-level info app:app >> $LOG_FILE 2>&1