Update app.py

app.py

@@ -18,8 +18,8 @@ def get_first_subword(word):
     except:
         return tokenizer(word, add_special_tokens=False)['input_ids'][0]
 
-def search(word_to_lookup, num_neighbors=20):
-    i = get_first_subword(word_to_lookup)
+def search(token_to_lookup, num_neighbors=100):
+    i = get_first_subword(token_to_lookup)
     _ , I = index.search(input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])
     return hits.values