Added predict and load_model functions.