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')
|