christopher commited on
Commit
79ae41f
Β·
1 Parent(s): 23020da

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -25,7 +25,7 @@ def get_first_subword(word):
25
  except:
26
  return tokenizer(word, add_special_tokens=False)['input_ids'][0]
27
 
28
- def search(token_to_lookup, num_neighbors):
29
  i = get_first_subword(token_to_lookup)
30
  _ , I = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
31
  hits = lookup_table.take(I[0])
 
25
  except:
26
  return tokenizer(word, add_special_tokens=False)['input_ids'][0]
27
 
28
+ def search(token_to_lookup, num_neighbors=100):
29
  i = get_first_subword(token_to_lookup)
30
  _ , I = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
31
  hits = lookup_table.take(I[0])