Only new tokens will now be enclosed by asterisks, lowered the upper bound on edits for faster compute.

app.py

@@ -160,10 +160,13 @@ def unmask(text, temp, rounds):
         while new_token == masked[0]:
             if unsuccessful_iters > 5:
                 break
-            print(new_token)
+            print('skipped', new_token)
             new_token = sample_output(out, sampling='uniform')
             unsuccessful_iters += 1
-        split_text[mask_pos] = '*' + new_token + '*'
+        if new_token == masked[0]:
+            split_text[mask_pos] = new_token
+        else:
+            split_text[mask_pos] = '*' + new_token + '*'
         text = ' '.join(split_text)
 
     text = list(text)
@@ -173,8 +176,8 @@ def unmask(text, temp, rounds):
 
 textbox = gr.Textbox(label="Example prompts", lines=5)
 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')
+temp_slider = gr.Slider(1.0, 3.0, value=1.0, label='Creativity')
+edit_slider = gr.Slider(1, 20, step=5, value=1.0, label='Number of edits')
 
 demo = gr.Interface(
     fn=unmask,