Model Predict Function In Python