Update app.py

app.py

@@ -35,8 +35,13 @@ def id_replace_complex(s, threshold=0.2):
   
   masked = [s[:s.index(t)] + '<mask>' + s[s.index(t)+len(t):] for t in compl_tok]
   cands = fill_pipe(masked)
-  replacements = [el['token_str'][1:] if type(el) == dict else el[0]['token_str'][1:] for el in cands]
-  
+  # structure is different in 1 vs n complex words
+  replacements = [el['token_str'] if type(
+      el) == dict else el[0]['token_str'] for el in cands]
+  # some tokens get prefixed with space
+  replacements = [tok if tok.find(' ') == -1 else tok[1:]
+                  for tok in replacements]
+
   for i, el in enumerate(compl_tok):
     idx = s.index(el)
     s = s[:idx] + replacements[i] + s[idx+len(el):]