Upload gec_model.py (#1)

- Upload gec_model.py (a054769d1cd4d149db1357e29ff211fb4186ecf5)


Co-authored-by: An Thành Long <Cnydo@users.noreply.huggingface.co>

gec_model.py

@@ -89,11 +89,11 @@ 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
         self.resolve_cycles = resolve_cycles
-
         assert (
             chunk_size > 0 and chunk_size // 2 >= overlap_size
         ), "Chunk merging required overlap size must be smaller than half of chunk size"