plano/model_server/app/install.py

from load_models import (
    load_transformers,
    load_ner_models,
    load_toxic_model,
    load_jailbreak_model,
)

print("installing transformers")
load_transformers()
print("installing ner models")
load_ner_models()
print("installing toxic models")
load_toxic_model()
print("installing jailbreak models")
load_jailbreak_model()