halyn commited on
Commit
9106f28
·
1 Parent(s): bbfcd34

return qa_chain

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -60,7 +60,7 @@ def setup_qa_chain():
60
  print(f"Error loading model: {e}")
61
  return
62
  llm = HuggingFacePipeline(pipeline=pipe)
63
- qa_chain = load_qa_with_sources_chain(llm)
64
 
65
  # 메인 페이지 UI
66
  def main_page():
 
60
  print(f"Error loading model: {e}")
61
  return
62
  llm = HuggingFacePipeline(pipeline=pipe)
63
+ qa_chain = load_qa_chain(llm, chain_type="stuff")
64
 
65
  # 메인 페이지 UI
66
  def main_page():