Update README.md

README.md

@@ -37,10 +37,10 @@ docs_input = tokenizer(documents , padding="max_length", 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(
-document_input_ids=docs_input.input_ids.unsqueeze(0),
-document_attention_mask=docs_input.attention_mask.unsqueeze(0),
-question_input_ids=question_input.input_ids,
-question_attention_mask=question_input.attention_mask
+    document_input_ids=docs_input.input_ids.unsqueeze(0),
+    document_attention_mask=docs_input.attention_mask.unsqueeze(0),
+    question_input_ids=question_input.input_ids,
+    question_attention_mask=question_input.attention_mask
 )
 print("reranking scores: ", output.logits)
 ```