Update README.md
Browse files
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)
|