MaximEremeev commited on
Commit
e9e2747
·
verified ·
1 Parent(s): ef066a2

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -0
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