Update README.md

README.md

@@ -79,7 +79,7 @@ def predict_paraphrase(sentence1, sentence2):
         outputs = model(**inputs)
         logits = outputs.logits
         probs = torch.softmax(logits, dim=1)
-        paraphrase_prob = probs[0][1].item()   # assuming label 1 = paraphrase
+        paraphrase_prob = probs[0][1].item()
     return {"Paraphrase": paraphrase_prob, "Not Paraphrase": 1 - paraphrase_prob}
 
 ```