update Dockerfile

Dockerfile

@@ -8,11 +8,12 @@ RUN pip install --no-cache-dir --upgrade -r /requirements.txt
 
 RUN curl -fsSL https://ollama.com/install.sh | sh 
 
-RUN ollama pull phi2
-
 COPY start.sh /start.sh
 RUN chmod +x /start.sh
 
+COPY pull_models.sh /pull_models.sh
+RUN chmod +x /pull_models.sh
+
 EXPOSE 7860
 
 CMD ["/start.sh"]

pull_models.sh//

@@ -0,0 +1,9 @@
+#!/bin/bash
+
+# Kiểm tra xem model đã được tải chưa
+if [ ! -d "/root/.ollama/models/manifests/phi2" ]; then
+  echo "Pulling phi2 model..."
+  ollama pull phi2
+else
+  echo "phi2 model already exists"
+fi

start.sh

@@ -6,5 +6,8 @@ ollama serve &
 # Wait for Ollama to start
 sleep 5
 
+# Pull models if needed
+/pull_models.sh
+
 # Start FastAPI app
 uvicorn app:app --host 0.0.0.0 --port 7860