Spaces:
Running
Running
Update app.py
Browse files
app.py
CHANGED
|
@@ -310,6 +310,7 @@ async def read_root(request: Request):
|
|
| 310 |
async def restore_text(req: RestoreRequest) -> Dict[str, Any]:
|
| 311 |
try:
|
| 312 |
is_char = req.mode == "char"
|
|
|
|
| 313 |
bpe_mask = tokenizer_bpe.mask_token
|
| 314 |
text = req.text.replace("#", "[GAP]")
|
| 315 |
|
|
|
|
| 310 |
async def restore_text(req: RestoreRequest) -> Dict[str, Any]:
|
| 311 |
try:
|
| 312 |
is_char = req.mode == "char"
|
| 313 |
+
mask = "[MASK]" if is_char else tokenizer_bpe.mask_token
|
| 314 |
bpe_mask = tokenizer_bpe.mask_token
|
| 315 |
text = req.text.replace("#", "[GAP]")
|
| 316 |
|