Spaces:
Sleeping
Sleeping
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 |
|