Ctaake commited on
Commit
97a1afd
·
verified ·
1 Parent(s): 8378078

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -44,6 +44,7 @@ def inference(message, history, systemPrompt=SYSTEM_PROMPT, temperature=0.9, max
44
  for stream_part in stream:
45
  partial_response += stream_part.token.text
46
  yield partial_response
 
47
 
48
 
49
  myAdditionalInputs = [
 
44
  for stream_part in stream:
45
  partial_response += stream_part.token.text
46
  yield partial_response
47
+ return partial_response.stripsuffix("<|END_OF_TURN_TOKEN|>")
48
 
49
 
50
  myAdditionalInputs = [