Update README.md

README.md

@@ -18,10 +18,10 @@ LFQA DPR-based retriever (vblagoje/dpr-question_encoder-single-lfqa-wiki and vbl
 
 
 ```python
-device = ("cuda" if torch.cuda.is_available() else "cpu")
+from transformers import DPRContextEncoder, DPRContextEncoderTokenizer
+
 model = DPRQuestionEncoder.from_pretrained("vblagoje/dpr-question_encoder-single-lfqa-wiki").to(device)
 tokenizer = AutoTokenizer.from_pretrained("vblagoje/dpr-question_encoder-single-lfqa-wiki")
-_ = model.eval()
 
 input_ids = tokenizer("Why do airplanes leave contrails in the sky?", return_tensors="pt")["input_ids"]
 embeddings = model(input_ids).pooler_output