File size: 305 Bytes
bf9a369 |
1 2 3 4 5 6 7 8 9 |
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)
|