Spaces:
Running
Running
Update app.py
Browse files
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'] =
|
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})
|