Pavel Soriano
commited on
Commit
·
a0bc241
1
Parent(s):
647da32
updated Readme
Browse files
README.md
CHANGED
@@ -109,17 +109,17 @@ embeddings = model.forward(input_ids).pooler_output
|
|
109 |
print(embeddings)
|
110 |
```
|
111 |
|
112 |
-
And with `haystack
|
113 |
```
|
114 |
-
retriever = DensePassageRetriever(
|
115 |
-
|
116 |
-
|
117 |
-
|
118 |
-
|
119 |
-
|
120 |
-
|
121 |
-
)
|
122 |
```
|
|
|
123 |
## Acknowledgments
|
124 |
|
125 |
This work was performed using HPC resources from GENCI–IDRIS (Grant 2020-AD011011224).
|
|
|
109 |
print(embeddings)
|
110 |
```
|
111 |
|
112 |
+
And with `haystack`, we use it as a retriever:
|
113 |
```
|
114 |
+
retriever = DensePassageRetriever(
|
115 |
+
document_store=document_store,
|
116 |
+
query_embedding_model="etalab-ia/dpr-question_encoder-fr_qa-camembert",
|
117 |
+
passage_embedding_model="etalab-ia/dpr-ctx_encoder-fr_qa-camembert",
|
118 |
+
model_version=dpr_model_tag,
|
119 |
+
infer_tokenizer_classes=True,
|
120 |
+
)
|
|
|
121 |
```
|
122 |
+
|
123 |
## Acknowledgments
|
124 |
|
125 |
This work was performed using HPC resources from GENCI–IDRIS (Grant 2020-AD011011224).
|