Update app.py

app.py

@@ -31,7 +31,7 @@ def get_first_subword(word):
     except:
         return tokenizer(word, add_special_tokens=False)['input_ids'][0]
 
-def search(token_to_lookup, num_neighbors=500):
+def search(token_to_lookup, num_neighbors):
     i = get_first_subword(token_to_lookup)
     _ , I = index_L2_unnormalized.search(unnormalized_input_embeddings[i:i+1], num_neighbors)
     hits = lookup_table.take(I[0])
@@ -43,8 +43,8 @@ def search(token_to_lookup, num_neighbors=500):
 iface = gr.Interface(
     fn=search,
     
-    #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..."),
+    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 tokens"), gr.Textbox(label="Nearest subwords")],
     examples=[
         ["##logy"],