Update app.py

app.py

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