christopher commited on
Commit
9fa01fd
Β·
1 Parent(s): 19c7d3c

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=100):
29
  i = get_first_subword(token_to_lookup)
30
  _ , I_IP = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
31
  hits_IP = lookup_table.take(I_IP[0])
 
25
  except:
26
  return tokenizer(word, add_special_tokens=False)['input_ids'][0]
27
 
28
+ def search(token_to_lookup, num_neighbors=500):
29
  i = get_first_subword(token_to_lookup)
30
  _ , I_IP = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
31
  hits_IP = lookup_table.take(I_IP[0])