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 |
|