Can anyone give an example code regarding how to predict using the trained model? Got stuck while trying to visualize something...