Update app.py

app.py

@@ -332,6 +332,9 @@ async def restore_text(req: RestoreRequest) -> Dict[str, Any]:
         is_char = req.mode == "char"
         mask    = "[MASK]" if is_char else tokenizer_bpe.mask_token
         text    = req.text.replace("#", "[GAP]")
+        # Lowering everything except special tokens
+        parts = SPECIAL_RE.split(text)
+        text = "".join(p if SPECIAL_RE.fullmatch(p) else p.lower() for p in parts if p)
 
         n_gaps = text.count("-") + text.count("[GAP]")