random2222 commited on
Commit
b0ad4e0
·
verified ·
1 Parent(s): 2971dc0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -46,7 +46,7 @@ def create_qa_system():
46
  return RetrievalQA.from_chain_type(
47
  llm=qa_pipeline,
48
  chain_type="stuff",
49
- retriever=db.as_retriever(search_kwargs={"k": 2})
50
  except Exception as e:
51
  raise gr.Error(f"Initialization failed: {str(e)}")
52
 
 
46
  return RetrievalQA.from_chain_type(
47
  llm=qa_pipeline,
48
  chain_type="stuff",
49
+ retriever=db.as_retriever(search_kwargs={"k": 2}))
50
  except Exception as e:
51
  raise gr.Error(f"Initialization failed: {str(e)}")
52