Update app.py

app.py

@@ -18,10 +18,17 @@ def get_first_subword(word):
     except:
         return tokenizer(word, add_special_tokens=False)['input_ids'][0]
 
-word_to_lookup = "##lity"
-num_neighbors = 20
-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])
-hits.values
+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 hits.values
+
+
+iface = gr.Interface(
+    fn=search,
+    inputs=gr.Textbox(lines=1, placeholder="Enter token..."),
+    outputs=gr.Textbox(label="Results"),
+)
+iface.launch(enable_queue=True, debug=True, show_error=True)