captain-awesome commited on
Commit
0076790
·
1 Parent(s): beb5508

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -1
app.py CHANGED
@@ -283,7 +283,8 @@ def retrieve_bot_answer(query, loaded_documents):
283
  """
284
  qa_bot_instance = create_retrieval_qa_bot(loaded_documents)
285
  bot_response = qa_bot_instance({"question": query})
286
- return bot_response
 
287
 
288
 
289
  # from your_module import load_model, set_custom_prompt, set_custom_prompt_condense, create_vector_database, retrieve_bot_answer
 
283
  """
284
  qa_bot_instance = create_retrieval_qa_bot(loaded_documents)
285
  bot_response = qa_bot_instance({"question": query})
286
+ answer = bot_response['answer']
287
+ return answer
288
 
289
 
290
  # from your_module import load_model, set_custom_prompt, set_custom_prompt_condense, create_vector_database, retrieve_bot_answer