pustozerov's picture
Implemented punctuation and capitalization into the Streamlit interface.
bf9a369
raw
history blame
305 Bytes
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)