venkat-srinivasan-nexusflow commited on
Commit
cdc924d
β€’
1 Parent(s): 1c4f8f5

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -430,12 +430,12 @@ class RavenDemo(gr.Blocks):
430
  summary_model_prompt, **SUMMARY_MODEL_GENERATION_KWARGS
431
  )
432
  for s in stream:
433
- s = s.removesuffix("<|end_of_turn|>")
434
  for c in s:
435
  summary_model_summary += c
436
  summary_model_summary = (
437
  summary_model_summary.lstrip().removesuffix(
438
- "<|end_of_turn|>"
439
  )
440
  )
441
  yield get_returns()
 
430
  summary_model_prompt, **SUMMARY_MODEL_GENERATION_KWARGS
431
  )
432
  for s in stream:
433
+ s = s.removesuffix("</s>")
434
  for c in s:
435
  summary_model_summary += c
436
  summary_model_summary = (
437
  summary_model_summary.lstrip().removesuffix(
438
+ "</s>"
439
  )
440
  )
441
  yield get_returns()