Update app.py

app.py

@@ -74,7 +74,8 @@ def chat_gen(message, history, return_buffer=True):
                 'question': message,
                 'context': retrieved_docs
         }
-        for token in conv_chain.stream(state):
+        ai_msg = conv_chain.invoke(state)
+        for token in ai_msg["answer"]:
             buffer += token
             yield buffer