Update app.py

app.py

@@ -56,7 +56,7 @@ concise_text = dataset["concise"]["text"]
 concise_text_string = "".join(concise_text)
 
 # Chunk and index the documents
-chunks = chunk_text(concise_text_string, chunk_size=300)  # chunks are 30 words each
+chunks = chunk_text(concise_text_string, chunk_size=450)
 # Build the vectorsore
 vectorstore = build_faiss_vectorstore(chunks)
 
@@ -67,7 +67,7 @@ async def chat(request: ChatRequest):
             client = genai.Client(api_key=google_api_key)
 
             # Retrieve relevant text
-            results = retrieve(request.message, vectorstore)
+            results = retrieve(request.message, vectorstore, top_k=10)
             formatted_results = "\n\n".join(results)
 
             rag_prompt = f"""You have access to the following relevant information retrieved based on the user's query: