updated docker files

Author: ArnabChatterjee20k

docker-compose.yml

@@ -28,17 +28,15 @@ services:
     build:
       context: .
       dockerfile: ollama.dockerfile
+      args:
+        - MODELS: "embeddinggemma mistral"
     container_name: ollama
     ports:
       - "11434:11434"
     restart: unless-stopped
     # persistent for caching models across restarts and preloading
     volumes:
       - ollama_models:/root/.ollama
-    environment:
-      - MODELS=embeddinggemma
-      # duration to keep model in memory
-      - OLLAMA_KEEP_ALIVE=24h
     networks:
       - utopia
 

ollama.dockerfile

@@ -1,7 +1,7 @@
 FROM ollama/ollama:0.12.7
 
 # Preload specific models
-ENV MODELS="embeddinggemma"
+ARG MODELS="embeddinggemma"
 ENV OLLAMA_KEEP_ALIVE=24h
 
 # Pre-pull models at build time for Docker layer caching