Update README.md

README.md

@@ -76,9 +76,9 @@ def infer(txt: Union[str, List[str]]):
     # Inportant: take only last token!
     return embedding.get('last_hidden_state')[:,-1,:].numpy()
 
-list_of_contexts = [...]
+list_of_contexts: List[str] = [...]
 emb_contexts = infer(list_of_contexts)
-list_of_queries = [...]
+list_of_queries: List[str] = [...]
 emb_queries = infer(list_of_queries)
 
 # Important: take cosine distance!
@@ -104,9 +104,9 @@ retriever = pipeline('feature-extraction', 'cmarkea/bloomz-560m-retriever-v2')
 # Inportant: take only last token!
 infer = lambda x: [ii[0][-1] for ii in retriever(x)]
 
-list_of_contexts = [...]
+list_of_contexts: List[str] = [...]
 emb_contexts = np.concatenate(infer(list_of_contexts), axis=0)
-list_of_queries = [...]
+list_of_queries: List[str] = [...]
 emb_queries = np.concatenate(infer(list_of_queries), axis=0)
 
 # Important: take cosine distance!