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 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.