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.

Learn More