Spaces:
Running
on
Zero
Running
on
Zero
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,
|