Update app.py

app.py

@@ -27,7 +27,7 @@ def load_vector_db(zip_file_path, extract_path):
 
 # Function to augment prompt
 def augment_prompt(query, vectordb):
-    results = vectordb.similarity_search(query, k=20)
+    results = vectordb.similarity_search(query, k=10)
     source_knowledge = "\n".join([x.page_content for x in results])
     augmented_prompt = f"""
     You are an AI assistant. Use the context provided below to answer the question as comprehensively as possible.