Update README.md

README.md

@@ -181,7 +181,7 @@ with torch.no_grad():
     test_representation = torch.mean(training_outputs.hidden_states[-1].squeeze(), dim=0)
 
 # Calculate similarity scores
-similarity_score = cos(training_input, test_input)
+similarity_score = cos(training_representation, test_representation)
 ```
 
 ## 引用 Citation