Fix the model running (related to logging), and add predict method
Fix the model running (related to logging), and add predict method