firqaaa commited on
Commit
0b86dec
·
1 Parent(s): 7483df3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
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.retriever.dense import DensePassageRetriever
34
- retriever = DensePassageRetriever(document_store=document_store,
 
 
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,