maxidl commited on
Commit
78dca84
·
1 Parent(s): 771dd42
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -193,7 +193,7 @@ def generate(paper_text, review_template):
193
  generated_text = ""
194
  for new_text in streamer:
195
  generated_text += new_text
196
- yield generated_text
197
  # generated_ids = model.generate(
198
  # input_ids=input_ids,
199
  # max_new_tokens=4096,
 
193
  generated_text = ""
194
  for new_text in streamer:
195
  generated_text += new_text
196
+ yield generated_text.replace("<|eot_id|>", "")
197
  # generated_ids = model.generate(
198
  # input_ids=input_ids,
199
  # max_new_tokens=4096,