brian-yu-nexusflow commited on
Commit
b93548c
Β·
1 Parent(s): 2285cb3

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
app.py CHANGED
@@ -389,6 +389,7 @@ class RavenDemo(gr.Blocks):
389
  summary_model_prompt, **SUMMARY_MODEL_GENERATION_KWARGS
390
  )
391
  for s in stream:
 
392
  for c in s:
393
  summary_model_summary += c
394
  summary_model_summary = (
 
389
  summary_model_prompt, **SUMMARY_MODEL_GENERATION_KWARGS
390
  )
391
  for s in stream:
392
+ s = s.removesuffix("<|end_of_turn|>")
393
  for c in s:
394
  summary_model_summary += c
395
  summary_model_summary = (