Pavel Soriano
commited on
Commit
•
30a7899
1
Parent(s):
c0c88f0
Update README card
Browse files
README.md
CHANGED
@@ -102,14 +102,13 @@ print(embeddings)
|
|
102 |
|
103 |
And with `haystack` (using `transformers-3.3.1`), we use it as a retriever (**note that we reference it from a local path**):
|
104 |
```
|
105 |
-
retriever = DensePassageRetriever(
|
106 |
-
|
107 |
-
|
108 |
-
|
109 |
-
|
110 |
-
|
111 |
-
|
112 |
-
)
|
113 |
```
|
114 |
## Acknowledgments
|
115 |
|
|
|
102 |
|
103 |
And with `haystack` (using `transformers-3.3.1`), we use it as a retriever (**note that we reference it from a local path**):
|
104 |
```
|
105 |
+
retriever = DensePassageRetriever(
|
106 |
+
document_store=document_store,
|
107 |
+
query_embedding_model="etalab-ia/dpr-question_encoder-fr_qa-camembert",
|
108 |
+
passage_embedding_model="etalab-ia/dpr-ctx_encoder-fr_qa-camembert",
|
109 |
+
model_version=dpr_model_tag,
|
110 |
+
infer_tokenizer_classes=True,
|
111 |
+
)
|
|
|
112 |
```
|
113 |
## Acknowledgments
|
114 |
|