from nemo.collections.nlp.models import PunctuationCapitalizationModel


punctuation_capitalization_model = PunctuationCapitalizationModel.from_pretrained("punctuation_en_distilbert")


def punctuation_capitalization(text):
    return punctuation_capitalization_model.add_punctuation_capitalization(text)