Spaces:

vanwdai
/

dialogflowAPI

Sleeping

OnlyBiggg commited on Apr 23, 2025

Commit

af86a55

1 Parent(s): c6554d8

fix device cpu

Files changed (1) hide show

app/ner/services/ner.py CHANGED Viewed

@@ -17,7 +17,8 @@ class NER:
         self.pipeline = pipeline(task=settings.TASK_NAME,
                                  model=self.model,
                                  tokenizer=self.tokenizer,
-                                 accelerator=settings.ACCELERATOR)
     def predict(self, text: str, entity_tag: str = None):
         if self.pipeline is None:

         self.pipeline = pipeline(task=settings.TASK_NAME,
                                  model=self.model,
                                  tokenizer=self.tokenizer,
+                                #  accelerator=settings.ACCELERATOR,
+                                 device=settings.DEVICE)
     def predict(self, text: str, entity_tag: str = None):
         if self.pipeline is None: