Carlosito16 commited on
Commit
065bd26
1 Parent(s): aa7bffb

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -103,7 +103,7 @@ def load_llm_model():
103
 
104
 
105
  @st.cache_resource
106
- def load_conversational_qa_memory_retriever(llm_model):
107
  question_generator = LLMChain(llm=llm_model, prompt=CONDENSE_QUESTION_PROMPT)
108
  doc_chain = load_qa_chain(llm_model, chain_type="stuff", prompt = PROMPT)
109
  memory = ConversationBufferWindowMemory(k = 3, memory_key="chat_history", return_messages=True, output_key='answer')
 
103
 
104
 
105
  @st.cache_resource
106
+ def load_conversational_qa_memory_retriever():
107
  question_generator = LLMChain(llm=llm_model, prompt=CONDENSE_QUESTION_PROMPT)
108
  doc_chain = load_qa_chain(llm_model, chain_type="stuff", prompt = PROMPT)
109
  memory = ConversationBufferWindowMemory(k = 3, memory_key="chat_history", return_messages=True, output_key='answer')