Spaces:
Paused
Paused
modify run to invoke
Browse files
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 |
-
|
72 |
-
|
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 |
|