Sfoglia il codice sorgente

fix post process in unimernet (#4213)

liuhongen1234567 5 mesi fa
parent
commit
718c3a9c6c

+ 1 - 1
paddlex/inference/models/formula_recognition/processors.py

@@ -798,7 +798,7 @@ class UniMERNetDecode(object):
             for i in reversed(range(len(toks[b]))):
                 if toks[b][i] is None:
                     toks[b][i] = ""
-                toks[b][i] = toks[b][i].replace("", " ").strip()
+                toks[b][i] = toks[b][i].replace("Ġ", " ").strip()
                 if toks[b][i] in (
                     [
                         self.tokenizer.bos_token,