fix model dir

app/ner/services/ner.py

@@ -10,8 +10,8 @@ class NER:
     def load_model(self):
         from transformers import AutoTokenizer, AutoModelForTokenClassification, pipeline
 
-        self.tokenizer = AutoTokenizer.from_pretrained(self.model_dir, locals=True)
-        self.model = AutoModelForTokenClassification.from_pretrained(self.model_dir, locals=True)
+        self.tokenizer = AutoTokenizer.from_pretrained(self.model_dir, local_files_only=True)
+        self.model = AutoModelForTokenClassification.from_pretrained(self.model_dir, local_files_only=True)
         self.pipeline = pipeline(settings.TASK_NAME, model=self.model, tokenizer=self.tokenizer)
         
     def predict(self, text: str, entity_tag: str = None):