Load the model only when necessary