Update README.md

README.md

@@ -85,6 +85,7 @@ with torch.no_grad():
 # Calculate similarities
 scores = torch.matmul(q_reps, p_reps.T)
 print(scores)
+# tensor([[-0.0112,  0.3316,  0.2376]], device='cuda:0')
 ```
 
 # Limitations