Update Dockerfile

Dockerfile

@@ -1,14 +1,10 @@
-# Use the official Ollama image as the base
 FROM ollama/ollama:latest
 
-# Set the model name (using lightweight all-minilm)
-ARG OLLAMA_MODEL="all-minilm"
+WORKDIR app
 
-# Pull the model during build
-RUN ollama pull ${OLLAMA_MODEL}
+COPY pull-model.sh /app
+RUN chmod +x /pull-model.sh && /pull-model.sh
 
-# Expose the Ollama API port
 EXPOSE 11434
 
-# Start the server directly
 CMD ["ollama", "serve"]