Maslionok
/

sudo_pipelines

Model card Files Files and versions Community

Gleb Vinarskis commited on Feb 24

Commit

d7605b7

·

1 Parent(s): a77df7c

change

Files changed (1) hide show

floret_language_recognition.py +2 -1

floret_language_recognition.py CHANGED Viewed

@@ -8,7 +8,8 @@ class FloretPipeline:
     def __call__(self, text):
         output = self.model.predict(text, k=1)
-        language = output[0].replace("__label__", "")
         return language

     def __call__(self, text):
         output = self.model.predict(text, k=1)
+        language, value = output
+        language = language[0].replace("__label__", "")
         return language