Hugging Face's logo

Spaces:
Cricles
/
demo_detoxi
Running

App Files Community
Cricles commited on
Commit
0d3b9b0
·
verified ·
1 Parent(s): 80214a5

Update model_wrapper/fasttext_wrapper.py

Browse files
Files changed (1) hide show
  1. model_wrapper/fasttext_wrapper.py +1 -1
model_wrapper/fasttext_wrapper.py CHANGED
@@ -7,4 +7,4 @@ class FasttextWrapper(object):
7
  self.model = fasttext.load_model(FasttextWrapper.MODEL_PATH)
8
 
9
  def __call__(self, text: str) -> str:
10
- return self.model.predict(text)
 
7
  self.model = fasttext.load_model(FasttextWrapper.MODEL_PATH)
8
 
9
  def __call__(self, text: str) -> str:
10
+ return self.model.predict(text)[0][0]