Update README.md

README.md

@@ -96,7 +96,7 @@ The following example is based on the API Pipeline of the Transformers library.
 from transformers import pipeline
 
 reranker = pipeline(
-    task='feature-extraction',
+    task='text-classification',
     model='cmarkea/bloomz-560m-reranking',
     top_k=None
 )
@@ -114,11 +114,12 @@ similarities = reranker(
     ]
 )
 
+score_label_1 = [
+    next(item['score'] for item in entry if item['label'] == 'LABEL_1') 
+    for entry in similarities
+]
 contexts_reranked = sorted(
-    filter(
-        lambda x: x[0]['label'] == "LABEL_1",
-        zip(similarities, contexts)
-    ),
+    zip(score_label_1, contexts),
     key=lambda x: x[0],
     reverse=True
 )