Update README.md

README.md

@@ -34,8 +34,8 @@ documents = [
 
 question = "什么是怀孕最显著也是最早的信号？"
 
-question_input = tokenizer('question', return_tensors="pt")
-docs_input = tokenizer(documents , padding="max_length", return_tensors="pt")
+question_input = tokenizer(question, padding=True, return_tensors="pt")
+docs_input = tokenizer(documents, padding=True, return_tensors="pt")
 # document input shape should be [batch_size, num_docs, seq_len]
 # so if only input one sample of documents, add one dim by unsqueeze(0)
 output = model(