nguyenvulebinh
/

spelling-oov

Text2Text Generation

encoder-decoder

Inference Endpoints

Model card Files Files and versions Community

nguyenvulebinh commited on Dec 15, 2021

Commit

48687d1

•

1 Parent(s): 95d8e06

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -23,7 +23,7 @@ spell_model = EncoderDecoderModel.from_pretrained(model_name)
 def oov_spelling(word, num_candidate=1):
     result = []
-    inputs = spell_tokenizer([word])
     input_ids = inputs['input_ids']
     attention_mask = inputs['attention_mask']
     inputs = {

 def oov_spelling(word, num_candidate=1):
     result = []
+    inputs = spell_tokenizer([word.lower()])
     input_ids = inputs['input_ids']
     attention_mask = inputs['attention_mask']
     inputs = {