IST199655 commited on
Commit
66e4a39
·
1 Parent(s): ade03a7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -3
app.py CHANGED
@@ -124,9 +124,7 @@ def respond(
124
  response = ""
125
  for token in streamer:
126
  response += token
127
- if "[Chatbot]:" in response: # Only stream the part after "[Chatbot]:"
128
- assistant_response = response.split("[Chatbot]:", 1)[1].strip()
129
- yield assistant_response
130
  print(response)
131
 
132
 
 
124
  response = ""
125
  for token in streamer:
126
  response += token
127
+ yield response[len(prompt):].strip()
 
 
128
  print(response)
129
 
130