Pavel Soriano commited on
Commit
30a7899
1 Parent(s): c0c88f0

Update README card

Browse files
Files changed (1) hide show
  1. README.md +7 -8
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(document_store=document_store,
106
- query_embedding_model="./etalab-ia/dpr-question_encoder-fr_qa-camembert",
107
- passage_embedding_model="./etalab-ia/dpr-ctx_encoder-fr_qa-camembert",
108
- use_gpu=True,
109
- embed_title=False,
110
- batch_size=16,
111
- use_fast_tokenizers=False
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