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

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -36,7 +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
- 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
 
 
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