Update README.md

README.md

@@ -12,13 +12,15 @@ type ids are all 0-s.
 An example usage:
 
 ```python
-from transformers import AutoTokenizer, DPRQuestionEncoder
+from transformers import AutoTokenizer, DPRContextEncoder
 
 tokenizer = AutoTokenizer.from_pretrained("NAACL2022/spider-trivia-ctx-encoder")
-model = DPRQuestionEncoder.from_pretrained("NAACL2022/spider-trivia-ctx-encoder")
+model = DPRContextEncoder.from_pretrained("NAACL2022/spider-trivia-ctx-encoder")
 
-question = "Who is the villain in lord of the rings"
-input_dict = tokenizer(question, return_tensors="pt")
+title = "Sauron"
+context = "Sauron is the title character and main antagonist of J. R. R. Tolkien's \"The Lord of the Rings\"."
+
+input_dict = tokenizer(title, context, return_tensors="pt")
 del input_dict["token_type_ids"]
 
 outputs = model(**input_dict)