Hugging Face's logo

soheeyang
/
dpr-question_encoder-single-trivia-base

Feature Extraction Transformers PyTorch TensorFlow dpr Inference Endpoints
Model card Files Community
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
  ```