Update model_wrapper/fasttext_wrapper.py

model_wrapper/fasttext_wrapper.py

@@ -7,4 +7,5 @@ class FasttextWrapper(object):
         self.model = fasttext.load_model(FasttextWrapper.MODEL_PATH)
 
     def __call__(self, text: str) -> str:
-        return self.model.predict(text)[0][0]
+        label, prob=self.model.predict(text)
+        return label[0],prob