halyn commited on
Commit
b1960ab
·
1 Parent(s): 4c64b1f

modify run to invoke

Browse files
Files changed (1) hide show
  1. app.py +5 -4
app.py CHANGED
@@ -67,10 +67,11 @@ def get_response_from_model(prompt):
67
  return "QA chain is not initialized."
68
 
69
  docs = st.session_state.knowledge_base.similarity_search(prompt)
70
- print("docs:", docs)
71
- response = st.session_state.qa_chain.run(input_documents=docs, question=prompt)
72
- print("response2:", response)
73
-
 
74
  if "Helpful Answer:" in response:
75
  response = response.split("Helpful Answer:")[1].strip()
76
 
 
67
  return "QA chain is not initialized."
68
 
69
  docs = st.session_state.knowledge_base.similarity_search(prompt)
70
+ print("docs:", docs) # 이까진 됐는데
71
+ # Chain의 invoke() 메소드 사용
72
+ response = st.session_state.qa_chain.invoke(input_documents=docs, question=prompt)
73
+ print("response:", response)
74
+
75
  if "Helpful Answer:" in response:
76
  response = response.split("Helpful Answer:")[1].strip()
77