christopher commited on
Commit
bed8476
Β·
1 Parent(s): a147e52

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -36,6 +36,7 @@ def search(token_to_lookup, num_neighbors=200):
36
  _ , I = index_L2_unnormalized.search(unnormalized_input_embeddings[i:i+1], num_neighbors)
37
  hits = lookup_table.take(I[0])
38
  results = hits.values[1:]
 
39
  return [r for r in results if not "##" in r], [[r for r in results if "##" in r]]
40
 
41
 
 
36
  _ , I = index_L2_unnormalized.search(unnormalized_input_embeddings[i:i+1], num_neighbors)
37
  hits = lookup_table.take(I[0])
38
  results = hits.values[1:]
39
+ results = [r for r in results if not "unused" in r]
40
  return [r for r in results if not "##" in r], [[r for r in results if "##" in r]]
41
 
42