fixed dockerfile

Dockerfile

@@ -9,7 +9,10 @@ WORKDIR /app
 COPY --chown=user ./requirements.txt requirements.txt
 RUN pip install --no-cache-dir --upgrade -r requirements.txt
 
-RUN ollama run gemma2:2b
+# Ensure `ollama` is installed
+RUN python -m pip show ollama
 
-COPY --chown=user . /app
-CMD ["uvicorn", "app:app", "--host", "0.0.0.0", "--port", "7860"]
+COPY --chown=user ./startup.sh /app/startup.sh
+RUN chmod +x /app/startup.sh
+
+CMD ["/app/startup.sh"]

startup.sh

@@ -0,0 +1,7 @@
+#!/bin/bash
+
+# Pull the Ollama model
+python -m ollama run gemma2:2b
+
+# Start the application
+exec uvicorn app:app --host 0.0.0.0 --port 7860