Fixed simple replacement logic to account for deletion of grammer (bread. is now the same as bread)

app.py

@@ -177,13 +177,13 @@ def unmask(text, temp, rounds):
         out = {item["token_str"]: item["score"] for item in res}
         new_token = sample_output(out, sampling)
         unsuccessful_iters = 0
-        while new_token == masked:
+        while masked in new_token:
             if unsuccessful_iters > 5:
                 break
             print('skipped', new_token)
             new_token = sample_output(out, sampling=sampling)
             unsuccessful_iters += 1
-        if new_token == masked:
+        if masked in new_token:
             split_text[mask_pos] = new_token
         else:
             split_text[mask_pos] = '*' + new_token + '*'