Update app.py

app.py

@@ -25,7 +25,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=250):
     i = get_first_subword(token_to_lookup)
     _ , I_IP = index_IP.search(normalized_input_embeddings[i:i+1], num_neighbors)
     hits_IP = lookup_table.take(I_IP[0])
@@ -43,9 +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.Slider(minimum=0, maximum=1000, value=250, step=10,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.Slider(minimum=0, maximum=1000, value=250, step=10,label="number of neighbors")],
+    inputs=gr.Textbox(lines=1, label="Vocabulary Token", placeholder="Enter token..."),
     outputs=[gr.Textbox(label="IP-Nearest tokens"), gr.Textbox(label="IP-Nearest subwords"), gr.Textbox(label="L2-Nearest tokens"), gr.Textbox(label="L2-Nearest subwords")],
     examples=[
         ["##logy"],