ollama_serve.py

import os
import sys

# get model name
model = sys.argv[1]

# start the server
os.system('ollama serve &')
# pull the model
os.system(f'ollama pull {model}')
# verify the contents
os.system(f'ollama pull {model}')