Spaces:

vanwdai
/

dialogflowAPI

Sleeping

OnlyBiggg commited on Apr 23, 2025

Commit

8519780

1 Parent(s): af86a55

fix accele

Files changed (1) hide show

app/ner/services/ner.py CHANGED Viewed

@@ -17,7 +17,6 @@ class NER:
         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):

         self.pipeline = pipeline(task=settings.TASK_NAME,
                                  model=self.model,
                                  tokenizer=self.tokenizer,
                                  device=settings.DEVICE)
     def predict(self, text: str, entity_tag: str = None):