|
|
@ -28,6 +28,8 @@ def predict(input, max_length, top_p, temperature, history=None):
|
|
|
|
|
|
|
|
|
|
|
|
with container:
|
|
|
|
with container:
|
|
|
|
if len(history) > 0:
|
|
|
|
if len(history) > 0:
|
|
|
|
|
|
|
|
if len(history)>MAX_BOXES:
|
|
|
|
|
|
|
|
history = history[-MAX_TURNS:]
|
|
|
|
for i, (query, response) in enumerate(history):
|
|
|
|
for i, (query, response) in enumerate(history):
|
|
|
|
message(query, avatar_style="big-smile", key=str(i) + "_user")
|
|
|
|
message(query, avatar_style="big-smile", key=str(i) + "_user")
|
|
|
|
message(response, avatar_style="bottts", key=str(i))
|
|
|
|
message(response, avatar_style="bottts", key=str(i))
|
|
|
@ -66,4 +68,4 @@ if 'state' not in st.session_state:
|
|
|
|
if st.button("发送", key="predict"):
|
|
|
|
if st.button("发送", key="predict"):
|
|
|
|
with st.spinner("AI正在思考,请稍等........"):
|
|
|
|
with st.spinner("AI正在思考,请稍等........"):
|
|
|
|
# text generation
|
|
|
|
# text generation
|
|
|
|
st.session_state["state"] = predict(prompt_text, max_length, top_p, temperature, st.session_state["state"])
|
|
|
|
st.session_state["state"] = predict(prompt_text, max_length, top_p, temperature, st.session_state["state"])
|
|
|
|