Update Dockerfile

Dockerfile

@@ -1,28 +1,3 @@
-# Use an official Python base image
-FROM python:3.8-slim
+FROM ollama/ollama
 
-# Set the working directory inside the container
-WORKDIR /app
-
-# Install system dependencies, curl, and other necessary utilities
-RUN apt-get update && apt-get install -y \
-    curl \
-    && pip install --upgrade pip
-
-# Install Ollama
-RUN curl -sSL https://ollama.com/install.sh | bash
-
-# Expose the port for the model server (assuming default port 5000)
-EXPOSE 5000
-
-# Start Ollama service, pull the model, and then serve it
-CMD /bin/bash -c "
-  # Start the Ollama service in the foreground
-  ollama & \
-  # Wait for Ollama to be ready
-  sleep 10 && \
-  # Pull the llama-3.2 model
-  ollama pull llama-3.2 && \
-  # Start serving the model
-  ollama serve --model llama-3.2 --port 5000
-"
+COPY ./pull-llama3.sh /pull-llama3.sh