Update app.py
Browse files
app.py
CHANGED
|
@@ -135,6 +135,7 @@ def respond(
|
|
| 135 |
gen_text = tokenizer.decode(gen_tokens[0])
|
| 136 |
print(gen_text)
|
| 137 |
gen_text= gen_text.replace(input_str,'')
|
|
|
|
| 138 |
|
| 139 |
yield gen_text
|
| 140 |
|
|
|
|
| 135 |
gen_text = tokenizer.decode(gen_tokens[0])
|
| 136 |
print(gen_text)
|
| 137 |
gen_text= gen_text.replace(input_str,'')
|
| 138 |
+
gen_text= gen_text.replace('<|END_OF_TURN_TOKEN|>','')
|
| 139 |
|
| 140 |
yield gen_text
|
| 141 |
|