increase k to 10

app.py

@@ -19,13 +19,14 @@ cross_encoder = CrossEncoder('cross-encoder/ms-marco-MiniLM-L-6-v2')
 documentdb = Simiandb("mystore", embedding_function=hf, mode="a")
 
 def search(query):
-    hits = documentdb.similarity_search(query)
+    hits = documentdb.similarity_search(query, k=10)
     cross_inp = [[query, hit] for hit in hits]
     cross_scores = cross_encoder.predict(cross_inp)
     hits = [hit for _, hit in sorted(zip(cross_scores, hits), reverse=True)]
-    return hits[0]
+    return hits
 
 iface = gr.Interface(fn=search, inputs="text", outputs="text")
 iface.launch()
 
-#print(search("what is the balloon boy hoax"))
+#print(search("what is the balloon boy hoax"))
+# print(search("date of birth of elon musk"))