HarnithaS commited on
Commit
ee9df87
·
1 Parent(s): dbf32ab
Files changed (1) hide show
  1. app.py +1 -2
app.py CHANGED
@@ -119,8 +119,7 @@ def generate_answer(user_query, context_documents):
119
  conversation_prompt = ChatPromptTemplate.from_template(PROMPT_TEMPLATE)
120
  response_chain = conversation_prompt | LANGUAGE_MODEL
121
  response = response_chain.invoke({"user_query": user_query, "document_context": context_text})
122
- print(response.keys())
123
- return response["answer"]
124
 
125
  # UI Configuration
126
 
 
119
  conversation_prompt = ChatPromptTemplate.from_template(PROMPT_TEMPLATE)
120
  response_chain = conversation_prompt | LANGUAGE_MODEL
121
  response = response_chain.invoke({"user_query": user_query, "document_context": context_text})
122
+ return response
 
123
 
124
  # UI Configuration
125