cmarkea
/

bloomz-3b-reranking

text-classification

text-generation-inference

Model card Files Files and versions

Cyrile commited on Apr 11, 2024

Commit

1ee5a9c

·

verified ·

1 Parent(s): d2db686

Update README.md

Files changed (1) hide show

README.md +2 -2

README.md CHANGED Viewed

@@ -99,7 +99,7 @@ reranker = pipeline(
     top_k=None
 )
-similarity = reranker(
     [
         dict(
             text=context, # the model was trained with context in `text`
@@ -111,7 +111,7 @@ similarity = reranker(
 contexts_reranked = sorted(
     filter(
         lambda x: x[0]['label'] == "LABEL_1",
-        zip(similarity, context_list)
     ),
     key=lambda x: x[0]
 )

     top_k=None
 )
+similarities = reranker(
     [
         dict(
             text=context, # the model was trained with context in `text`
 contexts_reranked = sorted(
     filter(
         lambda x: x[0]['label'] == "LABEL_1",
+        zip(similarities, contexts)
     ),
     key=lambda x: x[0]
 )