Update README.md

README.md

@@ -10,6 +10,7 @@ widget:
 This is the model used for knowledge clustering where we feed title-body pair and the classifier predicts if the pair is valid or not.
 For further information, please refer to https://github.com/yctam/dstc10_track2_task2 for the Github repository.
 
+Credit: Jiakai Zou, Wilson Tam
 
 ---
 ```python
@@ -20,6 +21,7 @@ def single_test(tokenizer, title_body_pair):
     model.eval()
     outputs = model(**result)
     predictions = outputs.logits.argmax(dim=-1)
+    # There was a mistake in flipping the labels.
     return True if predictions == 0 else False
 
 if __name__ == '__main__':