update main.py

main.py

@@ -97,7 +97,9 @@ def main():
         response = generate_response(user_input)
         st.session_state.messages.append({"role": "assistant", "content": response})
         with st.chat_message("assistant"):
-            st.write_stream(response_generator(response))
+            for word in response_generator(response):
+                st.write(word, end="")
+                st.write('')
 
     chatlog = format_messages_for_summary(st.session_state['messages'])
     st.sidebar.download_button(