firqaaa commited on
Commit
0402674
·
1 Parent(s): cf5ea83

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +5 -5
README.md CHANGED
@@ -41,10 +41,10 @@ datasets:
41
  ### Usage
42
 
43
  ```python
44
- from transformers import DPRQuestionEncoder, DPRQuestionEncoderTokenizer
45
 
46
- tokenizer = DPRQuestionEncoderTokenizer.from_pretrained('firqaaa/indo-dpr-question_encoder-multiset-base')
47
- model = DPRQuestionEncoder.from_pretrained('firqaaa/indo-dpr-question_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,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-question_encoder-multiset-base",
60
- passage_embedding_model="firqaaa/indo-dpr-question_encoder-multiset-base",
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,