Update app.py

app.py

@@ -12,8 +12,8 @@ unnormalized_input_embeddings = np.load("unnormalized_bert_input_embeddings.npy"
 index_L2 = IndexFlatL2(unnormalized_input_embeddings.shape[-1])  
 index_L2.add(unnormalized_input_embeddings)
 
-index_IP_normalized = IndexFlatIP(normalized_input_embeddings.shape[-1])  
-index_IP_normalized.add(normalized_input_embeddings)
+index_IP = IndexFlatIP(normalized_input_embeddings.shape[-1])  
+index_IP.add(normalized_input_embeddings)
 
 
 vocab = {v:k for k,v in tokenizer.vocab.items()}
@@ -27,7 +27,7 @@ def get_first_subword(word):
 
 def search(token_to_lookup, num_neighbors):
     i = get_first_subword(token_to_lookup)
-    _ , I = index_IP_normalized.search(normalized_input_embeddings[i:i+1], num_neighbors)
+    _ , I = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])
     results = hits.values[1:]
     results = [r for r in results if not "[unused" in r]