Update utils.py

utils.py

@@ -454,7 +454,7 @@ NER Interface : We Use this interface to infer sentence Time-Date tags.
 
 class NER(object):
 
-    def __init__(self, model_path, model_name, tags):
+    def __init__(self, model_path, tags):
         
         self.tags = tags
         self.device = "cuda" if torch.cuda.is_available() else "cpu"
@@ -462,7 +462,7 @@ class NER(object):
         roberta_path = "xlm-roberta-base"
         self.model = NERModel(n_labels=len(self.tags), roberta_path=roberta_path).to(self.device)
         # Load Fine-Tuned model
-        state_dict = torch.load(os.path.join(model_path, model_name))
+        state_dict = torch.load(model_path)
         self.model.load_state_dict(state_dict, strict=False)
         # Enable Evaluation mode
         self.model.eval()