src/utils/config.py

MODEL = None
TOKENIZER = None

def get_model():
    global MODEL, TOKENIZER

    if MODEL is None:
        print("Lazy loading model...")
        from src.pipeline.load_model import load_llm_model
        MODEL, TOKENIZER = load_llm_model()
        print("Model loaded!")

    return MODEL, TOKENIZER