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 |
|