Update gec_model.py (#1)

- Update gec_model.py (6088b6da2148be9a30c7468f7a54cb468ade726c)


Co-authored-by: Trinh Nhat Tuyen <NhatTuyen@users.noreply.huggingface.co>

gec_model.py

@@ -89,6 +89,7 @@ class GecBERTModel(torch.nn.Module):
         self.lowercase_tokens = lowercase_tokens
         self.min_error_probability = min_error_probability
         self.vocab = Vocabulary.from_files(vocab_path)
+        self.incorr_index = self.vocab.get_token_index("INCORRECT", "d_tags")
         self.log = log
         self.iterations = iterations
         self.confidence = confidence