Hugging Face's logo

nateraw
/
guesslang

Text Classification
generic
Model card Files Community
nateraw commited on
Commit
231ea7d
•
1 Parent(s): c1da753

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -5,4 +5,4 @@ class PreTrainedPipeline():
5
  self.guess = Guess()
6
 
7
  def __call__(self, text):
8
- return [{k: v} for k, v in self.guess.probabilities(text)[:5]]
 
5
  self.guess = Guess()
6
 
7
  def __call__(self, text):
8
+ return [{k: v} for k, v in self.guess.probabilities(text)[:5]]