Cyrile commited on
Commit
311c6e8
·
verified ·
1 Parent(s): babbc6b

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -2
README.md CHANGED
@@ -93,13 +93,20 @@ The following example utilizes the API Pipeline of the Transformers library.
93
  ```python
94
  from transformers import pipeline
95
 
96
- reranker = pipeline('feature-extraction', 'cmarkea/bloomz-3b-retriever', top_k=None)
 
 
 
 
97
 
98
  similarity = reranker(
99
  [dict(text=ii, text_pair=query) for ii in context_list]
100
  )
101
  context_reranked = sorted(
102
- filter(lambda x: x[0]['label'] == "LABEL_1", zip(similarity, context_list)),
 
 
 
103
  key=lambda x: x[0]
104
  )
105
  score, context_cleaned = zip(
 
93
  ```python
94
  from transformers import pipeline
95
 
96
+ reranker = pipeline(
97
+ task='feature-extraction',
98
+ model='cmarkea/bloomz-3b-retriever',
99
+ top_k=None
100
+ )
101
 
102
  similarity = reranker(
103
  [dict(text=ii, text_pair=query) for ii in context_list]
104
  )
105
  context_reranked = sorted(
106
+ filter(
107
+ lambda x: x[0]['label'] == "LABEL_1",
108
+ zip(similarity, context_list)
109
+ ),
110
  key=lambda x: x[0]
111
  )
112
  score, context_cleaned = zip(