Downloads · 30 days
72
0% of all-time downloads
SJ-Ray/Re-Punctuate
Re-Punctuate is a machine learning model from SJ-Ray. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers. The card lists the license as apache-2.0.
Downloads · 30 days
72
0% of all-time downloads
All-time downloads
15.5K
Public
Repo size
243 MB
Likes
12
Public
Click a slice to open those files.
.h5242 MB · 100%
From the Hugging Face model README
Re-Punctuate is a T5 model that attempts to correct Capitalization and Punctuations in the sentences.
<h3>DataSet:</h3>DialogSum dataset (115056 Records) was used to fine-tune the model for Punctuation and Capitalization correction.
<h3>Usage:</h3> <pre> from transformers import T5Tokenizer, TFT5ForConditionalGeneration tokenizer = T5Tokenizer.from_pretrained('SJ-Ray/Re-Punctuate') model = TFT5ForConditionalGeneration.from_pretrained('SJ-Ray/Re-Punctuate') input_text = 'the story of this brave brilliant athlete whose very being was questioned so publicly is one that still captures the imagination' inputs = tokenizer.encode("punctuate: " + input_text, return_tensors="tf") result = model.generate(inputs) decoded_output = tokenizer.decode(result[0], skip_special_tokens=True) print(decoded_output) </pre> <h4> Example: </h4> <b>Input:</b> the story of this brave brilliant athlete whose very being was questioned so publicly is one that still captures the imagination <br> <b>Output:</b> The story of this brave, brilliant athlete, whose very being was questioned so publicly, is one that still captures the imagination. <h4> Connect on: <a href="https://www.linkedin.com/in/suraj-kumar-710382a7" target="_blank">LinkedIn : Suraj Kumar</a></h4>