Spaces:

bigscience-data
/

token-explorer

Build error

christopher commited on Aug 18, 2022

Commit

857d106

1 Parent(s): d9adf81

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -19,7 +19,6 @@ def get_first_subword(word):
         return tokenizer(word, add_special_tokens=False)['input_ids'][0]
 def search(word_to_lookup, num_neighbors=20):
-    word_to_lookup = "##lity"
     i = get_first_subword(word_to_lookup)
     _ , I = index.search(input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])

         return tokenizer(word, add_special_tokens=False)['input_ids'][0]
 def search(word_to_lookup, num_neighbors=20):
     i = get_first_subword(word_to_lookup)
     _ , I = index.search(input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])