Update README.md

README.md

@@ -41,7 +41,7 @@ solution = ["To convert a string to an integer in Python 3, we use the built-in
             "Looking at the options, we can see that the correct function is option E: int(x [,base]).",
             "The answer is (E)."]
 input_text = question + ' \n\n' + ' \n\n\n\n'.join(solution) + ' \n\n\n\n' # solution steps are separated by ' \n\n\n\n'
-input_id = torch.tensor([self.tokenizer.encode(input_text)]).to(self.device)
+input_id = torch.tensor([tokenizer.encode(input_text)]).to(device)
 
 with torch.no_grad():
     logits = model(input_id).logits[:,:,candidate_tokens]