Require minimum number of specified edits to be made and added error message.

app.py

@@ -137,7 +137,10 @@ scrambler = pipeline("temp-scale", model="anferico/bert-for-patents")
 
 def unmask(text, temp, rounds):
     sampling = 'multi'
-    for round in range(rounds):
+    successful_iters = 0
+    unsuccessful_iters = 0
+    while successful_iters < rounds or unsuccessful_iters > 5:
+        unsuccessful_iters += 1
         tp = add_mask(text, size=1)
         masked_text, masked = tp[0], tp[1]
         split_text = masked_text.split()
@@ -156,6 +159,10 @@ def unmask(text, temp, rounds):
             continue
         split_text[mask_pos] = '*' + new_token + '*'
         text = ' '.join(split_text)
+        successful_iters += 1
+        unsuccessful_iters -= 1
+    if unsuccessful_iters > 5:
+        text = "Ran into an issue :( Please try again."
     text = list(text)
     text[0] = text[0].upper()
     return ''.join(text)