Hugging Face's logo

Spaces:
Cricles
/
demo_detoxi
Running

App Files Community
Cricles commited on
Commit
f3afcef
·
verified ·
1 Parent(s): 2b92000

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
@@ -8,4 +8,4 @@ class FasttextWrapper(object):
8
 
9
  def __call__(self, text: str) -> str:
10
  label, prob=self.model.predict(text)
11
- return label[0],prob
 
8
 
9
  def __call__(self, text: str) -> str:
10
  label, prob=self.model.predict(text)
11
+ return label[0],prob[0]