Update app.py

samadpls · Aug 19, 2024 · d2629e4 · d2629e4
1 parent d708727
commit d2629e4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/src/app.py b/src/app.py
@@ -112,7 +112,7 @@ def main():
                 st.write(prompt)
 
             prompt_template = PromptTemplate(
-                template=TEMPLATE, input_variables=["input",st.session_state.messages]
+                template=TEMPLATE, input_variables=["input"]
             )
 
             if "model" in st.session_state:
@@ -124,7 +124,7 @@ def main():
                 if st.session_state.messages[-1]["role"] != "assistant":
                     with st.chat_message("assistant"):
                         with st.spinner("Generating..."):
-                            response = llm_chain.invoke(prompt)
+                            response = llm_chain.invoke("input": prompt,)
                             st.markdown(response)
                             message = {
                                 "role": "assistant",