Spaces:
Build error
Build error
Commit ·
d0a445c
1
Parent(s): b016be0
Update app.py
Browse files
app.py
CHANGED
|
@@ -22,7 +22,7 @@ def search(token_to_lookup, num_neighbors=100):
|
|
| 22 |
i = get_first_subword(token_to_lookup)
|
| 23 |
_ , I = index.search(input_embeddings[i:i+1], num_neighbors)
|
| 24 |
hits = lookup_table.take(I[0])
|
| 25 |
-
return hits.values
|
| 26 |
|
| 27 |
|
| 28 |
iface = gr.Interface(
|
|
|
|
| 22 |
i = get_first_subword(token_to_lookup)
|
| 23 |
_ , I = index.search(input_embeddings[i:i+1], num_neighbors)
|
| 24 |
hits = lookup_table.take(I[0])
|
| 25 |
+
return hits.values[1:]
|
| 26 |
|
| 27 |
|
| 28 |
iface = gr.Interface(
|