refactor: replace load_model with get_model for improved model loading ea52793 abreza commited on Jun 11