Modify Bayes initializer to take the corpus directly