Update app.py

app.py

@@ -48,10 +48,10 @@ with gr.Blocks() as demo:
     
     df_output = gr.Dataframe(label="Similarity Results", wrap=True)
     
-    def search(query):
+    def search(query, num_results):
         query_embedding = model.encode([query], convert_to_tensor=True)
         return perform_similarity_search(query_embedding, embeddings, space_ids, descriptions, top_k=num_results)
     
-    input.submit(search, inputs=input, outputs=df_output)
+    input.submit(search, inputs=[input, num_results], outputs=df_output)
 
 demo.launch()