diff --git a/Dockerfile b/Dockerfile
index e662ce2..78edbcc 100644
--- a/Dockerfile
+++ b/Dockerfile
@@ -15,5 +15,6 @@ EXPOSE 8000
 
 ENV PYTHONUNBUFFERED=1
 
+RUN pip install ollama
 
 CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000", "--reload"]