Update freenlpc.py

AbdelrhmanNile · Sep 5, 2022 · f236482 · f236482
1 parent 1e1af1d
commit f236482
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/freenlpc/freenlpc.py b/freenlpc/freenlpc.py
@@ -111,7 +111,7 @@ def classification(self, text: str, lables: list, multiclass: bool =True):
             except requests.exceptions.HTTPError:
                 self.__init_api()
 
-    def classification2(self, text: str):
+    def classification2(self, text: str, lables: list, multiclass: bool =True):
         """perform classification on a piece of text.
 
         Args:
@@ -127,7 +127,7 @@ def classification2(self, text: str):
         while True:
             try:
                 sleep(1)
-                response = self.__models[self.classification2.__name__].classification(text)
+                response = self.__models[self.classification2.__name__].classification(text, labels, multiclass)
                 result = []
                 for i in range(len(response['labels'])):
                     info = {}