espejelomar commited on
Commit
f1d0592
1 Parent(s): 633dc2b

Update pipeline.py

Browse files
Files changed (1) hide show
  1. pipeline.py +1 -1
pipeline.py CHANGED
@@ -33,4 +33,4 @@ class ImageClassificationPipeline():
33
  {"label": str(self.id2label[i]), "score": float(preds[i])}
34
  for i in range(len(preds))
35
  ]
36
- return sorted(labels, key=lambda tup: tup["score"], reverse=True)[: self.top_k]
33
  {"label": str(self.id2label[i]), "score": float(preds[i])}
34
  for i in range(len(preds))
35
  ]
36
+ return labels