Update pipeline.py
Browse files- 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]]
|