Spaces:
Sleeping
Sleeping
Mimi
commited on
Commit
·
32e3a9a
1
Parent(s):
3b7c309
Fixed typos
Browse files
agent.py
CHANGED
|
@@ -128,7 +128,7 @@ class Naomi:
|
|
| 128 |
)
|
| 129 |
|
| 130 |
def respond(self, user_input: dict, **kwargs):
|
| 131 |
-
"""
|
| 132 |
# user msg handling
|
| 133 |
format_user_input = self.model.tokenizer_.hf_tokenizer.apply_chat_template([user_input], tokenize=False, add_generation_prompt=False)
|
| 134 |
self.chat_history += format_user_input
|
|
|
|
| 128 |
)
|
| 129 |
|
| 130 |
def respond(self, user_input: dict, **kwargs):
|
| 131 |
+
""" Called during stream. """
|
| 132 |
# user msg handling
|
| 133 |
format_user_input = self.model.tokenizer_.hf_tokenizer.apply_chat_template([user_input], tokenize=False, add_generation_prompt=False)
|
| 134 |
self.chat_history += format_user_input
|
app.py
CHANGED
|
@@ -76,7 +76,7 @@ def open_chat_window(**kwargs):
|
|
| 76 |
st.session_state.messages.add_message(role='user', content=user_input)
|
| 77 |
# Display user message in chat message container
|
| 78 |
msgbox.chat_message("user").write(user_input)
|
| 79 |
-
response = st.session_state.naomi.
|
| 80 |
msgbox.chat_message('assistant').write_stream(st.session_state.naomi.gen(response))
|
| 81 |
st.session_state.messages.add_message(role='assistant', content=response)
|
| 82 |
|
|
|
|
| 76 |
st.session_state.messages.add_message(role='user', content=user_input)
|
| 77 |
# Display user message in chat message container
|
| 78 |
msgbox.chat_message("user").write(user_input)
|
| 79 |
+
response = st.session_state.naomi.respond(st.session_state.messages[-1])
|
| 80 |
msgbox.chat_message('assistant').write_stream(st.session_state.naomi.gen(response))
|
| 81 |
st.session_state.messages.add_message(role='assistant', content=response)
|
| 82 |
|