Spaces:
Build error
Build error
fixed typo
Browse files
app.py
CHANGED
|
@@ -47,7 +47,7 @@ def generate(
|
|
| 47 |
)
|
| 48 |
|
| 49 |
better_prompt = tokenizer.decode(outputs[0])
|
| 50 |
-
better_prompt = better_prompt.replace("<pad
|
| 51 |
return better_prompt
|
| 52 |
|
| 53 |
|
|
|
|
| 47 |
)
|
| 48 |
|
| 49 |
better_prompt = tokenizer.decode(outputs[0])
|
| 50 |
+
better_prompt = better_prompt.replace("<pad>", "").replace("</s>", "")
|
| 51 |
return better_prompt
|
| 52 |
|
| 53 |
|