Update app.py

app.py

@@ -42,7 +42,7 @@ def invoke(openai_api_key, youtube_url, process_video, prompt):
     else:
         vector_db = Chroma(persist_directory = CHROMA_DIR, embedding_function = OpenAIEmbeddings())
     llm = ChatOpenAI(model_name = MODEL_NAME, temperature = 0)
-    qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(search_kwargs = {"k": 5}), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
+    qa_chain = RetrievalQA.from_chain_type(llm, retriever = vector_db.as_retriever(search_kwargs = {"k": 3}), return_source_documents = True, chain_type_kwargs = {"prompt": QA_CHAIN_PROMPT})
     result = qa_chain({"query": prompt})
     print(result)
     return result["result"]