Update app.py

app.py

@@ -22,7 +22,7 @@ def search(token_to_lookup, num_neighbors=100):
     i = get_first_subword(token_to_lookup)
     _ , I = index.search(input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])
-    return hits.values
+    return hits.values[1:]
 
 
 iface = gr.Interface(