veghar commited on
Commit
4c53382
1 Parent(s): 0d90c03

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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, AutoModelForSeq2SeqLM
30
  tokenizer = AutoTokenizer.from_pretrained("veghar/spell_correct_bart_base")
31
- model = AutoModelForSeq2SeqLM.from_pretrained("veghar/spell_correct_bart_base")
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')