Update app.py

app.py

@@ -17,8 +17,9 @@ def my_pipe(text):
     emb = np.array(vo.embed(text, model="voyage-2", input_type="document").embeddings)
 
     similarities = cosine_similarity(embeddings, emb)
-    most_similar_index = np.argmax(similarities)
-    return all_quotes[most_similar_index]
+    most_similar_indices = np.argsort(similarities, axis=0)[-5:]
+    
+    return [all_quotes[i] for i in most_similar_indices.flatten()]
 
 def launch(input):
     out = my_pipe(input)