Update Model/NER/VLSP2021/Predict_Ner.py

Model/NER/VLSP2021/Predict_Ner.py

@@ -49,7 +49,7 @@ class ViTagger(object):
     def __init__(self, model_path: Union[str or os.PathLike],  no_cuda=False):
         self.device = 'cuda' if not no_cuda and torch.cuda.is_available() else 'cpu'
         print("[ViTagger] VnCoreNLP loading ...")
-        self.rdrsegmenter = VnCoreNLP("/VnCoreNLP/VnCoreNLP-1.1.1.jar", annotators="wseg", max_heap_size='-Xmx500m')
+        self.rdrsegmenter = VnCoreNLP("VnCoreNLP/VnCoreNLP-1.1.1.jar", annotators="wseg", max_heap_size='-Xmx500m')
         print("[ViTagger] Model loading ...")
         self.model, self.tokenizer,  self.max_seq_len, self.label2id, self.use_crf = self.load_model(model_path, device=self.device)
         self.id2label = {idx: label for idx, label in enumerate(self.label2id)}