Update demo.py

demo.py

@@ -124,10 +124,10 @@ prompt = PromptTemplate(
     template=template
 )
 chat_history= ""
-def answer(user_question):
+def answer(user_input):
     llm = GooglePalm(temperature=0, google_api_key=os.environ['PALM'])
     chain = LLMChain(llm=llm, prompt=prompt)
-    keywords = chain.run({'user_question':user_question, 'chat_history':chat_history})
+    keywords = chain.run({'user_question':user_input, 'chat_history':chat_history})
 
     topK=3
 #vectorize the query
@@ -152,7 +152,7 @@ def answer(user_question):
 )
 
     #ans2 = llm_chain.predict(full_result_string)
-    ans = llm_chain.predict(user_msg= f"{full_result_string} ---\n\n {prompt}")
+    ans = llm_chain.predict(user_question= f"{full_result_string} ---\n\n {user_input}")
 
     return ans