Added sanity checking for successful iterations

app.py

@@ -18,7 +18,7 @@ ex_str3 = "The graphite plane is composed of a two-dimensional hexagonal lattice
           "length and a width parallel to the graphite plane and a thickness orthogonal to the graphite plane with at " \
           "least one of the length, width, and thickness values being 100 nanometers or smaller. "
 
-examples = [[ex_str1, 1.2, 50],
+examples = [[ex_str1, 1.2, 1],
            [ex_str2, 1.5, 10],
            [ex_str3, 1.4, 20]]
 
@@ -134,8 +134,8 @@ scrambler = pipeline("temp-scale", model="anferico/bert-for-patents")
 
 def unmask(text, temp, rounds):
     sampling = 'multi'
-
-    for _ in range(rounds):
+    successful_iters = 0
+    for round in range(rounds):
         text = add_mask(text, size=1)
         split_text = text.split()
         res = scrambler(text, temp=temp, top_k=10)
@@ -153,13 +153,14 @@ def unmask(text, temp, rounds):
             continue
         split_text[mask_pos] = new_token
         text = ' '.join(split_text)
+        successful_iters += 1
     text = list(text)
     text[0] = text[0].upper()
-    return ''.join(text)
+    return ''.join(text) + str(successful_iters)
 
 
 textbox = gr.Textbox(label="Example prompts", lines=5)
-output_textbox = gr.Textbox(placeholder="", lines=4)
+output_textbox = gr.Textbox(placeholder="Output will appear here", lines=4)
 temp_slider = gr.Slider(1.0, 2.0, value=1.0, label='Creativity')
 edit_slider = gr.Slider(1, 150, step=5, value=1.0, label='Number of edits')