schuler commited on
Commit
f6f4325
·
verified ·
1 Parent(s): bb4dea4

Update app.py

Browse files

Don't strip result

Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -136,7 +136,7 @@ def get_response(system_message, chat_history, user_text, max_new_tokens=256, co
136
  st.session_state.last_response = generated_text
137
 
138
  # Extract the assistant's response
139
- assistant_response = generated_text[len(prompt):].strip()
140
 
141
  if continue_last:
142
  # Append the continued text to the last assistant message
 
136
  st.session_state.last_response = generated_text
137
 
138
  # Extract the assistant's response
139
+ assistant_response = generated_text[len(prompt):] # .strip()
140
 
141
  if continue_last:
142
  # Append the continued text to the last assistant message