github
/
ChatGLM2-6B
mirror of https://github.com/THUDM/ChatGLM2-6B
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

Merge 77e1bf7af1 into cb8e8b43c0

pull/358/merge
Weitian Leung 2024-09-03 14:42:40 +08:00 committed by GitHub
parent cb8e8b43c0 77e1bf7af1
commit 73de130bbf
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
1 changed files with 6 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

17
web_demo2.py
View File

@ -45,19 +45,14 @@ for i, (query, response) in enumerate(st.session_state.history):
st.markdown(query)
with st.chat_message(name="assistant", avatar="assistant"):
st.markdown(response)
with st.chat_message(name="user", avatar="user"):
input_placeholder = st.empty()
with st.chat_message(name="assistant", avatar="assistant"):
message_placeholder = st.empty()
prompt_text = st.text_area(label="用户命令输入",
height=100,
placeholder="请在这儿输入您的命令")
prompt_text = st.chat_input(placeholder="请在这儿输入您的命令")
if prompt_text:
with st.chat_message(name="user", avatar="user"):
st.markdown(prompt_text)
with st.chat_message(name="assistant", avatar="assistant"):
message_placeholder = st.empty()
button = st.button("发送", key="predict")
if button:
input_placeholder.markdown(prompt_text)
history, past_key_values = st.session_state.history, st.session_state.past_key_values
for response, history, past_key_values in model.stream_chat(tokenizer, prompt_text, history,
past_key_values=past_key_values,