Won't count iteration if no changes are made.

app.py

@@ -30,9 +30,10 @@ def add_mask(text, size=1):
     if '[MASK]' in split_text:
         return text
     idx = np.random.randint(len(split_text), size=size)
+    masked = split_text[idx]
     for i in idx:
         split_text[i] = '[MASK]'
-    return ' '.join(split_text)
+    return ' '.join(split_text), masked
 
 
 class TempScalePipe(FillMaskPipeline):
@@ -135,7 +136,7 @@ scrambler = pipeline("temp-scale", model="anferico/bert-for-patents")
 def unmask(text, temp, rounds):
     sampling = 'multi'
     for round in range(rounds):
-        text = add_mask(text, size=1)
+        text, masked = add_mask(text, size=1)
         split_text = text.split()
         res = scrambler(text, temp=temp, top_k=10)
         mask_pos = [i for i, t in enumerate(split_text) if 'MASK' in t][0]
@@ -148,7 +149,7 @@ def unmask(text, temp, rounds):
             idx = np.random.randint(0, len(score_list))
         score = score_list[idx]
         new_token = score_to_str[score]
-        if len(list(new_token)) < 2:
+        if len(list(new_token)) < 2 or new_token == masked:
             continue
         split_text[mask_pos] = '*' + new_token + '*'
         text = ' '.join(split_text)