farmax commited on
Commit
c1afe21
·
verified ·
1 Parent(s): bd456c0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -9,6 +9,7 @@ from langchain_community.vectorstores import Chroma
9
  from langchain_community.document_loaders import UnstructuredPDFLoader, PyPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain.chains import RetrievalQAWithSourcesChain
 
12
  from langchain.schema import Document
13
  from langchain.memory import ConversationBufferMemory
14
  from langchain.callbacks.manager import CallbackManager
@@ -76,7 +77,7 @@ def initialize_LLM(llm_option, llm_temperature, max_tokens, top_k, vector_db, pr
76
  print(f"Using default LLM {default_llm} for {language}")
77
  llm_name = default_llm
78
 
79
- qa_chain = load_qa_with_sources_from_chain_type(
80
  llm=llm_name,
81
  chain_type="stuff",
82
  retriever=vector_db.as_retriever(),
 
9
  from langchain_community.document_loaders import UnstructuredPDFLoader, PyPDFLoader
10
  from langchain.text_splitter import CharacterTextSplitter
11
  from langchain.chains import RetrievalQAWithSourcesChain
12
+ # from langchain.chains import load_qa_with_sources_from_chain_type
13
  from langchain.schema import Document
14
  from langchain.memory import ConversationBufferMemory
15
  from langchain.callbacks.manager import CallbackManager
 
77
  print(f"Using default LLM {default_llm} for {language}")
78
  llm_name = default_llm
79
 
80
+ qa_chain = RetrievalQAWithSourcesChain(
81
  llm=llm_name,
82
  chain_type="stuff",
83
  retriever=vector_db.as_retriever(),