SJ-Ray commited on
Commit
dd5a4b9
·
1 Parent(s): a3de7e0

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -14,8 +14,8 @@ DialogSum dataset (115056 Records) was used to fine-tune the model for Punctuati
14
  <pre>
15
 
16
  from transformers import T5Tokenizer, TFT5ForConditionalGeneration
17
- tokenizer = T5Tokenizer.from_pretrained('SJ-Ray/Re-Punctuate/')
18
- model = TFT5ForConditionalGeneration.from_pretrained('SJ-Ray/Re-Punctuate/')
19
  input_text = 'the story of this brave brilliant athlete whose very being was questioned so publicly is one that still captures the imagination'
20
  inputs = tokenizer.encode("punctuate: " + input_text, return_tensors="tf")
21
  result = model.generate(inputs)
 
14
  <pre>
15
 
16
  from transformers import T5Tokenizer, TFT5ForConditionalGeneration
17
+ tokenizer = T5Tokenizer.from_pretrained('SJ-Ray/Re-Punctuate')
18
+ model = TFT5ForConditionalGeneration.from_pretrained('SJ-Ray/Re-Punctuate')
19
  input_text = 'the story of this brave brilliant athlete whose very being was questioned so publicly is one that still captures the imagination'
20
  inputs = tokenizer.encode("punctuate: " + input_text, return_tensors="tf")
21
  result = model.generate(inputs)