Nyanfa commited on
Commit
9ddb3d7
·
verified ·
1 Parent(s): 0511f23

Add a process to remove the leading and trailing whitespace characters in Prefill

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -41,6 +41,7 @@ def get_ai_response(messages):
41
  st.session_state.response = ""
42
  shown_message = ""
43
 
 
44
  if st.session_state.prefill:
45
  messages.append({"role": "assistant", "content": st.session_state.prefill})
46
  st.session_state.response += st.session_state.prefill
 
41
  st.session_state.response = ""
42
  shown_message = ""
43
 
44
+ st.session_state.prefill = st.session_state.prefill.strip()
45
  if st.session_state.prefill:
46
  messages.append({"role": "assistant", "content": st.session_state.prefill})
47
  st.session_state.response += st.session_state.prefill