From db51c22093f8bdee48a5444dbbe4fb30cd2b1b98 Mon Sep 17 00:00:00 2001
From: Tanvir <tanvir.ahmed.tonoy@skiff.com>
Date: Tue, 9 Jan 2024 13:46:07 +0600
Subject: [PATCH] refactor(run-ollama-docker.sh): update with port variables

---
 run-ollama-docker.sh | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)

diff --git a/run-ollama-docker.sh b/run-ollama-docker.sh
index 89bf10be..c8ce166a 100644
--- a/run-ollama-docker.sh
+++ b/run-ollama-docker.sh
@@ -1,11 +1,14 @@
 #!/bin/bash
 
+host_port=11434
+container_port=11434
+
 read -r -p "Do you want ollama in Docker with GPU support? (y/n): " use_gpu
 
 docker rm -f ollama || true
 docker pull ollama/ollama:latest
 
-docker_args="-d -v ollama:/root/.ollama -p 11434:11434 --name ollama ollama/ollama"
+docker_args="-d -v ollama:/root/.ollama -p $host_port:$container_port --name ollama ollama/ollama"
 
 if [ "$use_gpu" == "y" ]; then
     docker_args+=" --gpus=all"