Illia56 commited on
Commit
6aac681
1 Parent(s): a4849de

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -2
app.py CHANGED
@@ -170,8 +170,9 @@ if prompt := textinput:
170
  st.chat_message("human",avatar = "🧑‍💻").markdown(prompt)
171
  st.session_state.messages.append({"role": "human", "content": prompt})
172
  with st.status("Requesting Client..."):
173
- response = st.session_state.qa.run(prompt)
 
174
  with st.chat_message("assistant", avatar='🦙'):
175
- st.markdown(response)
176
  # Add assistant response to chat history
177
  st.session_state.messages.append({"role": "assistant", "content": response})
 
170
  st.chat_message("human",avatar = "🧑‍💻").markdown(prompt)
171
  st.session_state.messages.append({"role": "human", "content": prompt})
172
  with st.status("Requesting Client..."):
173
+ response = st.session_state.qa(
174
+ {"query": prompt})
175
  with st.chat_message("assistant", avatar='🦙'):
176
+ st.markdown(response["result"]+'\n'+response["source_documents"])
177
  # Add assistant response to chat history
178
  st.session_state.messages.append({"role": "assistant", "content": response})