Update README.md
Browse files
README.md
CHANGED
@@ -26,9 +26,9 @@ This model works best for ''WORD-BASED'' spell correction(`not so good with the
|
|
26 |
|
27 |
## How to Get Started with the Model
|
28 |
```python
|
29 |
-
from transformers import AutoTokenizer,
|
30 |
tokenizer = AutoTokenizer.from_pretrained("veghar/spell_correct_bart_base")
|
31 |
-
model =
|
32 |
|
33 |
text='believ'
|
34 |
text_tok=tokenizer(text,padding=True, return_tensors='tf')
|
|
|
26 |
|
27 |
## How to Get Started with the Model
|
28 |
```python
|
29 |
+
from transformers import AutoTokenizer, TFBartForConditionalGeneration
|
30 |
tokenizer = AutoTokenizer.from_pretrained("veghar/spell_correct_bart_base")
|
31 |
+
model = TFBartForConditionalGeneration.from_pretrained("veghar/spell_correct_bart_base")
|
32 |
|
33 |
text='believ'
|
34 |
text_tok=tokenizer(text,padding=True, return_tensors='tf')
|