plot_model_prediction function

Plot model predictions