Update app.py

app.py

@@ -22,7 +22,8 @@ def search(token_to_lookup, num_neighbors=50):
     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[1:]
+    results = hits.values[1:]
+    return [r for r in results if not "##" in r], [[r for r in results if "##" in r]]
 
 
 iface = gr.Interface(
@@ -30,7 +31,7 @@ iface = gr.Interface(
     
     #inputs=[gr.Textbox(lines=1, label="Vocabulary Token", placeholder="Enter token..."), gr.Number(value=50, label="number of neighbors")],
     inputs=gr.Textbox(lines=1, label="Vocabulary Token", placeholder="Enter token..."),
-    outputs=gr.Textbox(label="Nearest subwords"),
+    outputs=[gr.Textbox(label="Nearest tokens"), gr.Textbox(label="Nearest subwords")]
     examples=[
         ["##logy"],
         ["##ness"],