Illia56 commited on
Commit
4e324e3
1 Parent(s): 7ac3460

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -149,7 +149,7 @@ if st.session_state.youtube_url and not st.session_state.setup_done :
149
  docsearch = FAISS.from_documents(docs, embeddings)
150
  retriever = docsearch.as_retriever()
151
  retriever.search_kwargs['distance_metric'] = 'cos'
152
- retriever.search_kwargs['k'] = 10
153
  with st.status("Running RetrievalQA..."):
154
  llama_instance = LlamaLLM()
155
  st.session_state.qa = RetrievalQA.from_chain_type(llm=llama_instance, chain_type="stuff", retriever=retriever,chain_type_kwargs={"prompt": prompt})
 
149
  docsearch = FAISS.from_documents(docs, embeddings)
150
  retriever = docsearch.as_retriever()
151
  retriever.search_kwargs['distance_metric'] = 'cos'
152
+ retriever.search_kwargs['k'] = 4
153
  with st.status("Running RetrievalQA..."):
154
  llama_instance = LlamaLLM()
155
  st.session_state.qa = RetrievalQA.from_chain_type(llm=llama_instance, chain_type="stuff", retriever=retriever,chain_type_kwargs={"prompt": prompt})