diff --git a/ai.sh b/ai.sh
index e4a911b..8120e78 100644
--- a/ai.sh
+++ b/ai.sh
@@ -1 +1,7 @@
-ollama run llama3:8b
+sudo systemctl start ollama
+if [ $1 = "llama" ]; then
+  ollama run llama3:8b
+else
+  ollama run gpt-oss:20b
+fi
+sudo systemctl stop ollama