Update README.md
Browse files
README.md
CHANGED
@@ -45,7 +45,7 @@ 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 tokoh utama dalam serial
|
49 |
embeddings = model(input_ids).pooler_output
|
50 |
```
|
51 |
|
|
|
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 tokoh utama dalam serial SLAMDunk?", return_tensors='pt')["input_ids"]
|
49 |
embeddings = model(input_ids).pooler_output
|
50 |
```
|
51 |
|