Update README.md

README.md

@@ -8,6 +8,8 @@ language:
 pipeline_tag: sentence-similarity
 ---
 
+
+
 ```python
 import numpy as np
 from transformers import pipeline
@@ -22,7 +24,10 @@ list_of_queries = [...]
 emb_queries = np.concatenate(infer(list_of_queries), axis=0)
 
 dist = cdist(emb_queries, emb_contexts, 'euclidean')
-top_k = lambda x: [[list_of_contexts[qq] for qq in ii] for ii in dist.argsort(axis=-1)[:,:x]]
+top_k = lambda x: [
+    [list_of_contexts[qq] for qq in ii]
+    for ii in dist.argsort(axis=-1)[:,:x]
+]
 # top 5 nearest contexts for each queries
 top_contexts = top_k(5)
 ```