Spaces:
Running
on
Zero
Running
on
Zero
Commit
•
327287c
1
Parent(s):
87a3951
sanity hack
Browse files
app.py
CHANGED
|
@@ -158,6 +158,8 @@ def model_inference(
|
|
| 158 |
# yield generated_text, image
|
| 159 |
print("after yield")
|
| 160 |
|
|
|
|
|
|
|
| 161 |
rendered_page = render_webpage(generated_text)
|
| 162 |
return generated_text, rendered_page
|
| 163 |
|
|
|
|
| 158 |
# yield generated_text, image
|
| 159 |
print("after yield")
|
| 160 |
|
| 161 |
+
# Sanity hack
|
| 162 |
+
generated_text = generated_text.replace("</s>", "")
|
| 163 |
rendered_page = render_webpage(generated_text)
|
| 164 |
return generated_text, rendered_page
|
| 165 |
|