fixing invoke

src/generator.py

@@ -18,7 +18,7 @@ def promt_template():
     prompt = ChatPromptTemplate.from_messages(
         [
             ("system",prompt_in_chat_format),
-            ("human", "{query}")
+            ("human", "{input}")
         ])
     #RAG_PROMPT_TEMPLATE = tokenizer.apply_chat_template(
     #prompt_in_chat_format, tokenize=False, add_generation_prompt=True)
@@ -41,6 +41,6 @@ def answer_with_rag(
     #print(final_prompt)
     # Redact an answer
     print("=> Generating answer...")
-    response=retrieval_chain.invoke(query)
+    response=retrieval_chain.invoke({'input':query})
 
     return response['answer'], response["context"]