Spaces:

johnnyfivefingers
/

answer-machine

Runtime error

johnnyfivefingers commited on Mar 28, 2023

Commit

739e3b5

•

1 Parent(s): afa1a50

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -32,7 +32,7 @@ def answer_question(document, question):
     similarity_scores = cosine_similarity(query_embedding, doc_embeddings)
     # Find the top 3 most similar documents
-    most_similar_idxs = similarity_scores.argsort()[0][-1:]
     # Extract the answers from the top 3 most similar documents
     answers = []

     similarity_scores = cosine_similarity(query_embedding, doc_embeddings)
     # Find the top 3 most similar documents
+    most_similar_idxs = similarity_scores.argsort()[0][-3:]
     # Extract the answers from the top 3 most similar documents
     answers = []