Update README.md
Browse files
README.md
CHANGED
|
@@ -41,10 +41,10 @@ datasets:
|
|
| 41 |
### Usage
|
| 42 |
|
| 43 |
```python
|
| 44 |
-
from transformers import
|
| 45 |
|
| 46 |
-
tokenizer =
|
| 47 |
-
model =
|
| 48 |
input_ids = tokenizer("Siapa nama pengarang manga Yu-Gi-Oh?", return_tensors='pt')["input_ids"]
|
| 49 |
embeddings = model(input_ids).pooler_output
|
| 50 |
```
|
|
@@ -56,8 +56,8 @@ from haystack.nodes import DensePassageRetriever
|
|
| 56 |
from haystack.document_stores import InMemoryDocumentStore
|
| 57 |
|
| 58 |
retriever = DensePassageRetriever(document_store=InMemoryDocumentStore(),
|
| 59 |
-
query_embedding_model="firqaaa/indo-dpr-
|
| 60 |
-
passage_embedding_model="firqaaa/indo-dpr-
|
| 61 |
max_seq_len_query=64,
|
| 62 |
max_seq_len_passage=256,
|
| 63 |
batch_size=16,
|
|
|
|
| 41 |
### Usage
|
| 42 |
|
| 43 |
```python
|
| 44 |
+
from transformers import DPRContextEncoder, DPRContextEncoderTokenizer
|
| 45 |
|
| 46 |
+
tokenizer = DPRContextEncoderTokenizer.from_pretrained('firqaaa/indo-dpr-ctx_encoder-multiset-base')
|
| 47 |
+
model = DPRContextEncoder.from_pretrained('firqaaa/indo-dpr-ctx_encoder-multiset-base')
|
| 48 |
input_ids = tokenizer("Siapa nama pengarang manga Yu-Gi-Oh?", return_tensors='pt')["input_ids"]
|
| 49 |
embeddings = model(input_ids).pooler_output
|
| 50 |
```
|
|
|
|
| 56 |
from haystack.document_stores import InMemoryDocumentStore
|
| 57 |
|
| 58 |
retriever = DensePassageRetriever(document_store=InMemoryDocumentStore(),
|
| 59 |
+
query_embedding_model="firqaaa/indo-dpr-ctx_encoder-multiset-base",
|
| 60 |
+
passage_embedding_model="firqaaa/indo-dpr-ctx_encoder-multiset-base",
|
| 61 |
max_seq_len_query=64,
|
| 62 |
max_seq_len_passage=256,
|
| 63 |
batch_size=16,
|