Update gec_model.py

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