Follow @bedapudi6788 Star Fork

Designed with ASR outputs in mind, DeepPunct uses LSTM encoder and decoders with Luong attention for automatic punctuation restoration.


# Install tensorflow or tensorflow-gpu separately
pip install deepcorrect


# if you are using gpu for prediction, please see for restricting memory usage

from deepcorrect import DeepCorrect
corrector = DeepCorrect('params_path', 'checkpoint_path')
corrector.correct('how are you')
# How are you?

The pre-trained models are available here.

Learn More