fix acceleb

app/ner/services/ner.py

@@ -16,8 +16,8 @@ class NER:
         self.model = ORTModelForTokenClassification.from_pretrained(self.model_dir)
         self.pipeline = pipeline(task=settings.TASK_NAME, 
                                  model=self.model, 
-                                 tokenizer=self.tokenizer, 
-                                 device=settings.DEVICE)
+                                 tokenizer=self.tokenizer,
+                                 accelerator=settings.ACCELERATOR)
         
     def predict(self, text: str, entity_tag: str = None):
         if self.pipeline is None: