Update Dockerfile

Dockerfile

@@ -1,12 +1,19 @@
-FROM python:3.9-slim
+# Use the official Ollama image as the base
+FROM ollama/ollama
 
-WORKDIR /app
+# Install Python and dependencies for the custom API
+RUN apt-get update && apt-get install -y python3 python3-pip
+RUN pip3 install fastapi uvicorn requests
 
-COPY requirements.txt .
-RUN pip install --no-cache-dir -r requirements.txt
+# Copy a script to pull the model at container startup
+COPY entrypoint.sh /entrypoint.sh
+RUN chmod +x /entrypoint.sh
 
-COPY . .
+# Copy the FastAPI app
+COPY app.py /app.py
 
+# Expose the port for the FastAPI server
 EXPOSE 8000
 
-CMD ["uvicorn", "main:app", "--host", "0.0.0.0", "--port", "8000"]
+# Run the entrypoint script and start the FastAPI server
+CMD ["/entrypoint.sh"]