Hugging Face's logo

soheeyang
/
dpr-question_encoder-single-trivia-base

Feature Extraction
Transformers
PyTorch
TensorFlow
dpr
Model card Files Community
1
soheeyang commited on
Commit
4a6b45f
·
1 Parent(s): 2801db1

Update readme

Browse files
Files changed (1) hide show
  1. README.md +1 -1
README.md CHANGED
@@ -36,6 +36,6 @@ from transformers import DPRQuestionEncoder, AutoTokenizer
36
  tokenizer = AutoTokenizer.from_pretrained("soheeyang/dpr-question_encoder-single-trivia-base")
37
  question_encoder = DPRQuestionEncoder.from_pretrained("soheeyang/dpr-question_encoder-single-trivia-base")
38
 
39
- data = tokenizer("context comes here", return_tensors="pt")
40
  question_embedding = question_encoder(**data).pooler_output # embedding vector for question
41
  ```
 
36
  tokenizer = AutoTokenizer.from_pretrained("soheeyang/dpr-question_encoder-single-trivia-base")
37
  question_encoder = DPRQuestionEncoder.from_pretrained("soheeyang/dpr-question_encoder-single-trivia-base")
38
 
39
+ data = tokenizer("question comes here", return_tensors="pt")
40
  question_embedding = question_encoder(**data).pooler_output # embedding vector for question
41
  ```