Model.predict Function In Python