Update README.md
Browse files
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(
|
|
|
|
|
|
|
|
|
97 |
|
98 |
similarity = reranker(
|
99 |
[dict(text=ii, text_pair=query) for ii in context_list]
|
100 |
)
|
101 |
context_reranked = sorted(
|
102 |
-
filter(
|
|
|
|
|
|
|
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(
|