Spaces:
Build error
Build error
Commit
Β·
9fa01fd
1
Parent(s):
19c7d3c
Update app.py
Browse files
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_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])
|