DeepPunct
Follow @bedapudi6788 Star Fork
Designed with ASR outputs in mind, DeepPunct uses LSTM encoder and decoders with Luong attention for automatic punctuation restoration.
Installation
# Install tensorflow or tensorflow-gpu separately
pip install deepcorrect
Usage
# if you are using gpu for prediction, please see https://stackoverflow.com/questions/34199233/how-to-prevent-tensorflow-from-allocating-the-totality-of-a-gpu-memory 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.