Update README.md

README.md

@@ -58,6 +58,7 @@ model = AutoModelForSequenceClassification.from_pretrained("Fan-s/reddit-tc-bert
 post = "don't make gravy with asbestos."
 response = "i'd expect someone with a culinary background to know that. since we're talking about school dinner ladies, they need to learn this pronto."
 
+# Predict whether the two sentences match 
 def predict(post, response, max_seq_length=128):
     with torch.no_grad():
         args =  (post, response)
@@ -67,8 +68,8 @@ def predict(post, response, max_seq_length=128):
         item = torch.argmax(logits, dim=1)
         predict_label = label_list[item]
         return predict_label, logits
-
-# predict whether the two sentences match                 
+         
 predict_label, logits = predict(post, response)
+# Matched
 print("predict_label:", predict_label)
 ```