random2222 commited on
Commit
d12de7f
·
verified ·
1 Parent(s): b1f884f

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ def create_qa_system():
54
  return RetrievalQA.from_chain_type(
55
  llm=llm,
56
  chain_type="stuff",
57
- retriever=db.as_retriever(search_kwargs={"k": 2})
58
  except Exception as e:
59
  raise gr.Error(f"Initialization failed: {str(e)}"))
60
 
 
54
  return RetrievalQA.from_chain_type(
55
  llm=llm,
56
  chain_type="stuff",
57
+ retriever=db.as_retriever(search_kwargs={"k": 2}))
58
  except Exception as e:
59
  raise gr.Error(f"Initialization failed: {str(e)}"))
60