Upload gec_model.py

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"