fixed bug where n was always 10 in nearest neighbours function

app.py

@@ -16,13 +16,15 @@ with gr.Blocks() as demo:
             inputs=[
                 gr.Textbox(label='Word 1 (required)', placeholder='χρηστήριον'),
                 gr.Radio(label='Time slice (required)', choices=["archaic_cbow", "classical_cbow", "early_roman_cbow", "hellen_cbow", "late_roman_cbow"]),
-                gr.Slider(label='Number of neighbours', minimum=1, maximum=50, step=1, value=10)
+                gr.Slider(label='Number of neighbours', minimum=1, maximum=50, step=1)
             ],
             outputs=gr.DataFrame(
                 label="Result",
-                headers=["Word", "Time slice", "Cosine similarity"]
+                headers=["Word", "Time slice", "Cosine similarity"],
+                type="pandas", visible=True, interactive=False
                 ),
-            submit_btn='Calculate'
+            submit_btn='Calculate',
+            
         )
 
     

word2vec.py

@@ -159,7 +159,7 @@ def get_nearest_neighbours(word, time_slice_model, n=10, models=load_all_models(
                     nearest_neighbours.remove(smallest_neighbour)
                     nearest_neighbours.append((word, model_name, cosine_similarity_vectors))
 
-    return sorted(nearest_neighbours, key=lambda x: x[2], reverse=True)[:10]
+    return sorted(nearest_neighbours, key=lambda x: x[2], reverse=True)