Update README.md
Browse files
README.md
CHANGED
|
@@ -30,8 +30,10 @@ embeddings = model(input_ids).pooler_output
|
|
| 30 |
You can use it using `haystack` as follows:
|
| 31 |
|
| 32 |
```
|
| 33 |
-
from haystack.
|
| 34 |
-
|
|
|
|
|
|
|
| 35 |
query_embedding_model="firqaaa/indo-dpr-passage_encoder-single-squad-base",
|
| 36 |
passage_embedding_model="firqaaa/indo-dpr-passage_encoder-single-squad-base",
|
| 37 |
max_seq_len_query=64,
|
|
|
|
| 30 |
You can use it using `haystack` as follows:
|
| 31 |
|
| 32 |
```
|
| 33 |
+
from haystack.nodes import DensePassageRetriever
|
| 34 |
+
from haystack.document_stores import InMemoryDocumentStore
|
| 35 |
+
|
| 36 |
+
retriever = DensePassageRetriever(document_store=InMemoryDocumentStore(),
|
| 37 |
query_embedding_model="firqaaa/indo-dpr-passage_encoder-single-squad-base",
|
| 38 |
passage_embedding_model="firqaaa/indo-dpr-passage_encoder-single-squad-base",
|
| 39 |
max_seq_len_query=64,
|